ott.solvers.was_solver.WassersteinSolver

class ott.solvers.was_solver.WassersteinSolver(linear_solver, threshold=0.001, min_iterations=5, max_iterations=50, store_inner_errors=False)

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

Parameters:

• linear_solver (Union[Sinkhorn, LRSinkhorn])

• threshold (float)

• min_iterations (int)

• max_iterations (int)

• store_inner_errors (bool)

Methods

Attributes

is_low_rank	Whether the solver is low-rank.
rank	Rank of the linear OT solver.