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.

setup_anndata(adata, *args, **kwargs)

Sets up the AnnData object for this model.

to_device(device)

Move model to device.

train()

Trains the model.