ott.problems.linear.potentials.EntropicPotentials.plot_ot_map#

EntropicPotentials.plot_ot_map(source, target, forward=True, ax=None, legend_kwargs=None, scatter_kwargs=None)#

Plot data and learned optimal transport map.

Parameters:

  • source – samples from the source measure

  • target – samples from the target measure

  • forward – use the forward map from the potentials if True, otherwise use the inverse map

  • ax – axis to add the plot to

  • scatter_kwargs – additional kwargs passed into scatter()

  • legend_kwargs – additional kwargs passed into legend()

Returns:

a matplotlib figure and axis with the plots