ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter

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

Continuous Wasserstein barycenter solver [Cuturi and Doucet, 2014].

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

init_state(bar_prob, bar_size[, x_init, rng])	Initialize the state of the Wasserstein barycenter iterations.
output_from_state(state)	param state

Attributes

is_low_rank

Whether the solver is low-rank.