tropea_clustering.plot.plot_sankey

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

Plots the Sankey diagram at the desired frames.

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

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

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

  • tmp_frame_list (List[int]) – The list of windows at which we want to plot the Sankey.

Notes

  • If there are no assigned window, we still need the “-1” state for consistency

  • Requires kaleido.