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
Whether the solver is low-rank.
Rank of the linear OT solver.