ott.neural.methods.expectile_neural_dual.ENOTPotentials.plot_ot_map

ENOTPotentials.plot_ot_map(source, target, samples=None, forward=True, ax=None, scatter_kwargs=None, legend_kwargs=None)

Plot data and learned optimal transport map.

Parameters:

• source (Array) – samples from the source measure

• target (Array) – samples from the target measure

• samples (Optional[Array]) – extra samples to transport, either source (if forward) or target (if not forward) by default.

• forward (bool) – use the forward map from the potentials if True, otherwise use the inverse map.

• ax (Optional[Axes]) – axis to add the plot to

• scatter_kwargs (Optional[Dict[str, Any]]) – additional kwargs passed into scatter()

• legend_kwargs (Optional[Dict[str, Any]]) – additional kwargs passed into legend()

Return type:

Tuple[Figure, Axes]

Returns:

Figure and axes.