moscot.problems.generic.GWProblem.prepare

GWProblem.prepare(key, x_attr=None, y_attr=None, policy='sequential', cost='sq_euclidean', cost_kwargs=mappingproxy({}), a=None, b=None, subset=None, reference=None, x_callback=None, y_callback=None, x_callback_kwargs=mappingproxy({}), y_callback_kwargs=mappingproxy({}))[source]

Prepare the individual quadratic subproblems.

See also

  • TODO(michalk8): add an example how to pass x_attr/y_attr.

Parameters:
Return type:

GWProblem[TypeVar(K, bound= Hashable), TypeVar(B, bound= OTProblem)]

Returns:

: Returns self and updates the following fields: