ott.tools.plot.Plot

Contents

ott.tools.plot.Plot#

class ott.tools.plot.Plot(fig=None, ax=None, threshold=-1.0, scale=200, show_lines=True, cmap='cool', scale_alpha_by_coupling=False, alpha=0.7, title=None, xlim=None, ylim=None)[source]#

Plot an optimal transport map between two point clouds.

This object can either plot or update a plot, to create animations as a FuncAnimation, which can in turned be saved to disk at will. There are two design principles here:

  1. we do not rely on saving to/loading from disk to create animations

  2. we try as much as possible to disentangle the transport problem from its visualization.

We use 2D scatter plots by default, relying on PCA visualization for d>3 data. This step requires a conversion to a numpy array, in order to compute leading singular values. This tool is therefore not designed having performance in mind.

Parameters:

  • fig (Optional[Figure]) – Specify figure object. Created by default

  • ax (Optional[Axes]) – Specify axes objects. Created by default

  • threshold (float) – value below which links in transportation matrix won’t be plotted. This value should be negative when using animations.

  • scale (int) – scale used for marker plots.

  • show_lines (bool) – whether to show OT lines, as described in ot.matrix argument

  • cmap (str) – color map used to plot line colors.

  • scale_alpha_by_coupling (bool) – use or not the coupling’s value as proxy for alpha

  • alpha (float) – default alpha value for lines.

  • title (Optional[str]) – title of the plot.

  • xlim (Optional[List[float]])

  • ylim (Optional[List[float]])

Methods

animate(transports[, titles, frame_rate])

Make an animation from several transports.

update(ot[, title])

Update a plot with a transport instance.
Contents