ott.solvers.was_solver.WassersteinSolver

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)[source]#

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

Parameters:

Methods

Attributes

is_low_rank

Whether the solver is low-rank.