ott.tools.conformal.OTCP.fit_transport

ott.tools.conformal.OTCP.fit_transport#

OTCP.fit_transport(x, y, epsilon=0.1, n_target=8192, rng=None, sampler_kwargs=None, **kwargs)[source]#

Fit the transport map.

Parameters:

  • x (Array) – Features of shape [n, dim_x] to fit the transport map.

  • y (Array) – Targets of shape [n, dim_y] to fit the transport map.

  • epsilon (float) – Epsilon regularization

  • n_target (int) – Total number of points used to create the target measure.

  • rng (Optional[Array]) – Random number generator.

  • sampler_kwargs (Optional[Any]) – Keyword arguments passed for the sampler.

  • kwargs (Any) – Keyword arguments for solve().

Return type:

OTCP

Returns:

Self and updates sinkhorn_output, offset, scale.