scvi.model.SCANVI.predict

SCANVI.predict(adata=None, indices=None, soft=False, batch_size=None)[source]

Return cell label predictions.

Parameters
adata : AnnData | NoneOptional[AnnData] (default: None)

AnnData object that has been registered via setup_anndata().

indices : Sequence[int] | NoneOptional[Sequence[int]] (default: None)

Return probabilities for each class label.

soft : boolbool (default: False)

If True, returns per class probabilities

batch_size : int | NoneOptional[int] (default: None)

Minibatch size for data loading into model. Defaults to scvi.settings.batch_size.

Return type

ndarray | DataFrameUnion[ndarray, DataFrame]