ott.solvers.was_solver.WassersteinSolver

class ott.solvers.was_solver.WassersteinSolver(epsilon=None, rank=-1, linear_ot_solver=None, min_iterations=5, max_iterations=50, threshold=0.001, store_inner_errors=False, **kwargs)

A generic solver for problems that use a linear problem in inner loop.

Parameters

• epsilon (Optional[float]) –

• rank (int) –

• linear_ot_solver (Union[Sinkhorn, LRSinkhorn, None]) –

• min_iterations (int) –

• max_iterations (int) –

• threshold (float) –

• store_inner_errors (bool) –

• kwargs (Any) –

Methods

Attributes

is_low_rank

Whether the solver is low-rank.