ott.tools.conformal.OTCP.fit_transport

ott.tools.conformal.OTCP.fit_transport#

OTCP.fit_transport(x, y, epsilon=0.1, n_target=8192, n_per_radius=None, rng=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 when sampling the target measure.

  • n_per_radius (Optional[int]) – Number of points selected in a given radius.

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

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

Return type:

OTCP

Returns:

Self and updates sinkhorn_output, offset, scale.

Contents