scvi.model.base.BaseModelClass¶

class scvi.model.base.BaseModelClass(adata=None)[source]¶

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.
`to_device`(device)	Move model to device.
`train`()	Trains the model.