ott.tools#
The tools package contains high level functions that build on outputs produced by core functions. They can be used to compute Sinkhorn divergences [Séjourné et al., 2019], instantiate transport matrices, provide differentiable approximations to ranks and quantile functions [Cuturi et al., 2019], etc.
Segmented Sinkhorn#
|
Compute regularized OT cost between subsets of vectors in x and y. |
Plotting#
|
Plot an optimal transport map between two point clouds. |
Sinkhorn Divergence#
Compute Sinkhorn divergence defined by a geometry, weights, parameters. |
|
Compute Sinkhorn divergence between subsets of vectors in x and y. |
Soft Sorting Algorithms#
|
Apply the soft quantile operator on the input tensor. |
|
Renormalize inputs so that its quantiles match those of targets/weights. |
|
Soft quantizes an input according using num_levels values along axis. |
|
Apply the soft rank operator on input tensor. |
|
Apply the soft sort operator on a given axis of the input. |
|
Sort a multidimensional array according to a real valued criterion. |
Clustering#
|
K-means clustering using Lloyd's algorithm [Lloyd, 1982]. |
|
Output of the |
ott.tools.gaussian_mixture package#
Gaussian Mixtures#
|
Normal distribution. |
|
Gaussian Mixture model. |
Coupled pair of Gaussian mixture models. |
|
|
Initialize a GMM via K-means++ with retries on failure. |
|
Fit a GMM using the EM algorithm. |
|
Get a function that performs penalized EM. |