moscot.problems.space.MappingProblem.impute

MappingProblem.impute(var_names=None, device=None, batch_size=None)

Impute the expression of specific genes.

Parameters:

  • var_names (Optional[Sequence[str]]) – Genes in var_names to impute. If None, use all genes in adata_sc.

  • device (Union[Literal['cpu', 'gpu', 'tpu'], str, None]) – Device where to transfer the solutions, see to().

  • batch_size (Optional[int]) – Number of features to process at once. If None, process all features at once. Larger values will require more memory.

Return type:

AnnData

Returns:

: Annotated data object with the imputed gene expression.