ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter#
- class ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter(epsilon=None, min_iterations=5, max_iterations=50, threshold=0.001, store_inner_errors=False, quad_solver=None, **kwargs)[source]#
Gromov-Wasserstein barycenter solver.
- Parameters:
min_iterations (
int
) – Minimum number of iterations.max_iterations (
int
) – Maximum number of outermost iterations.threshold (
float
) – Convergence threshold.store_inner_errors (
bool
) – Whether to store the errors of the GW solver, as well as its linear solver, at each iteration for each measure.quad_solver (
Optional
[GromovWasserstein
]) – The GW solver.kwargs (
Any
) – Keyword argument forGromovWasserstein
. Only used whenquad_solver = None
.
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.