ott.initializers.linear

ott.initializers.linear#

Initializers for linear OT problems, focusing on Sinkhorn and low-rank solvers.

Sinkhorn Initializers#

initializers.DefaultInitializer()

Default initialization of Sinkhorn dual potentials/primal scalings.

initializers.GaussianInitializer()

Gaussian initializer [Thornton and Cuturi, 2022].

initializers.SortingInitializer([...])

Sorting initializer [Thornton and Cuturi, 2022].

initializers.SubsampleInitializer(subsample_n_x)

Subsample initializer [Thornton and Cuturi, 2022].

Low-rank Sinkhorn Initializers#

initializers_lr.LRInitializer(rank, **kwargs)

Base class for low-rank initializers.

initializers_lr.RandomInitializer(rank, **kwargs)

Low-rank Sinkhorn factorization using random factors.

initializers_lr.Rank2Initializer(rank, **kwargs)

Low-rank Sinkhorn factorization using rank-2 factors [Scetbon et al., 2021].

initializers_lr.KMeansInitializer(rank[, ...])

K-means initializer for low-rank Sinkhorn [Scetbon and Cuturi, 2022].

initializers_lr.GeneralizedKMeansInitializer(rank)

Generalized k-means initializer [Scetbon and Cuturi, 2022].