tropea_clustering.plot.plot_state_populations

tropea_clustering.plot.plot_state_populations(title, n_windows, labels)[source]

Plot the populations of states over time.

Here’s an example of the output:

Example Image

For each trajectory frame, plots the fraction of the population of each cluster. “ENV0” refers to the unclassified data.

Parameters:
  • title (str) – The path of the .png file the figure will be saved as.

  • n_windows (int) – The number of windows used.

  • labels (ndarray of shape (n_particles * n_windows,)) – The output of the clustering algorithm.