moscot.base.problems.CondOTProblem.prepare¶
- CondOTProblem.prepare(policy_key, policy, xy, xx, conditions, a=None, b=None, subset=None, reference=None, **kwargs)[source]¶
Prepare conditional optimal transport problem.
- Parameters:
policy_key (
str
) – %(key)spolicy (
Literal
['sequential'
,'star'
,'external_star'
,'explicit'
,'triu'
,'tril'
]) – Policy defining which pairs of distributions to sample from during training.xy (
Mapping
[str
,Any
]) – Geometry defining the linear term. If passed as adict
,from_adata()
will be called.conditions (
Mapping
[str
,Any
]) – The description is missing.subset (
Optional
[Sequence
[Tuple
[TypeVar
(K
, bound=Hashable
),TypeVar
(K
, bound=Hashable
)]]]) – The description is missing.reference (
Optional
[TypeVar
(K
, bound=Hashable
)]) – The description is missing.**kwargs (
Any
) – The description is missing.
- Return type:
- Returns:
: Self and modifies the following attributes: TODO.