ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter

ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter#

class ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter(quadratic_solver, threshold=0.001, min_iterations=5, max_iterations=50, store_inner_errors=False)[source]#

Gromov-Wasserstein barycenter solver.

Parameters:

  • quadratic_solver (GromovWasserstein) – Quadratic OT solver.

  • threshold (float) – Convergence threshold.

  • min_iterations (int) – Minimum number of iterations.

  • max_iterations (int) – Maximum number of outermost iterations.

  • store_inner_errors (bool) – Whether to store the errors of the quadratic OT solver.

Methods

init_state(problem, bar_size[, bar_init, a, rng])

Initialize the (fused) Gromov-Wasserstein barycenter state.

output_from_state(state)

No-op.

update_state(state, iteration, problem[, ...])

Solve the (fused) Gromov-Wasserstein barycenter problem.

Attributes

is_low_rank

Whether the solver is low-rank.

rank

Rank of the linear OT solver.