moscot.problems.generic.GENOTLinProblem¶
- class moscot.problems.generic.GENOTLinProblem(adata, **kwargs)[source]¶
Class for solving Conditional Parameterized Monge Map problems / Conditional Neural OT problems.
Methods
cell_transition
(source, target[, ...])Aggregate the transport matrix.
compute_entropy
(source, target[, forward, ...])Compute the conditional entropy per cell.
compute_feature_correlation
(obs_key[, ...])Compute correlation of push-forward or pull-back distribution with features.
load
(path)Load the model from a file.
prepare
(key, joint_attr, conditional_attr[, ...])Prepare the
moscot.problems.generic.GENOTLinProblem
.pull
(source, target[, data, subset, ...])Pull mass from target to source.
push
(source, target[, data, subset, ...])Push mass from source to target.
save
(path[, overwrite])Save the problem to a file.
solve
([batch_size, seed, iterations, ...])Solve.
Attributes
Source annotated data object.
Batch key in
obs
.Collection of distributions.
Policy used to subset the data.
Kind of the underlying problem.
Solution of the optimal transport problem.
Solver of the optimal transport problem.
Problem stage.