scvi.model.base.BaseModelClass

class scvi.model.base.BaseModelClass(adata=None)

Abstract class for scvi-tools models.

Attributes

device
history	Returns computed metrics during training.
is_trained
test_indices
train_indices
validation_indices

Methods

load(dir_path[, adata, use_gpu])	Instantiate a model from the saved output.
save(dir_path[, overwrite, save_anndata])	Save the state of the model.
setup_anndata(adata, *args, **kwargs)	Sets up the AnnData object for this model.
to_device(device)	Move model to device.
train()	Trains the model.

previous

scvi.distributions.ZeroInflatedNegativeBinomial.zi_probs

next

scvi.model.base.BaseModelClass.device