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 invar_names
to impute. IfNone
, use all genes inadata_sc
.device (
Union
[Literal
['cpu'
,'gpu'
,'tpu'
],str
,None
]) – Device where to transfer the solutions, seeto()
.batch_size (
Optional
[int
]) – Number of features to process at once. IfNone
, process all features at once. Larger values will require more memory.
- Return type:
- Returns:
: Annotated data object with the imputed gene expression.