ott.tools.plot.Plot#

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

Plot an optimal transport map between two point clouds.

It enables to either plot or update a plot in a single object, offering the possibilities 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.

Methods

animate(transports[, frame_rate])

Make an animation from several transports.

update(ot)

Update a plot with a transport instance.

Parameters: