tropea_clustering.plot.plot_output_multi

tropea_clustering.plot.plot_output_multi(title, input_data, state_list, labels, delta_t)[source]

Plot a cumulative figure showing trajectories and identified states.

Parameters:

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

  • input_data (ndarray of shape (n_dims, n_particles, n_frames)) – The input data array.

  • state_list (List[StateMulti]) – The list of the cluster states.

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

  • delta_t (int) – The length of the signal sequences used.

  • plotted (All the data are)

  • been (colored according to the cluster thay have)

  • ellipses (assigned to. The clusters are shown as black)

  • and (whose orizontal)

  • Gaussians (vertical axis length is given by the standard deviation of the)

  • in (corresponding to the cluster. Unclassififed data points are colored)

  • purple.