ott.solvers.linear.continuous_barycenter.FreeBarycenterState

class ott.solvers.linear.continuous_barycenter.FreeBarycenterState(costs=None, linear_convergence=None, errors=None, x=None, a=None)

Holds the state of the Wasserstein barycenter solver.

Parameters:

• costs (Array | None) – Holds the sequence of regularized GW costs seen through the outer loop of the solver.

• linear_convergence (Array | None) – Holds the sequence of bool convergence flags of the inner Sinkhorn iterations.

• errors (Array | None) – Holds sequence of vectors of errors of the Sinkhorn algorithm at each iteration.

• x (Array | None) – barycenter points.

• a (Array | None) – barycenter weights.

Methods

count(value, /)	Return number of occurrences of value.
index(value[, start, stop])	Return first index of value.
set(**kwargs)	Return a copy of self, possibly with overwrites.
update(iteration, bar_prob, ...)	Update the state of the solver.

Attributes

a	Alias for field number 4
costs	Alias for field number 0
errors	Alias for field number 2
linear_convergence	Alias for field number 1
x	Alias for field number 3