ott.solvers.linear.continuous_barycenter.FreeBarycenterState

ott.solvers.linear.continuous_barycenter.FreeBarycenterState#

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

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, linear_solver, ...)

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
Contents