scgen.SCGEN.load

classmethod SCGEN.load(dir_path, adata=None, use_gpu=None, prefix=None, backup_url=None)

Instantiate a model from the saved output.

Parameters

dir_path : str

Path to saved outputs.

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

AnnData organized in the same way as data used to train model. It is not necessary to run setup_anndata, as AnnData is validated against the saved scvi setup dictionary. If None, will check for and load anndata saved with the model.

use_gpu : str | int | bool | NoneUnion[str, int, bool, None] (default: None)

Load model on default GPU if available (if None or True), or index of GPU to use (if int), or name of GPU (if str), or use CPU (if False).

prefix : str | NoneOptional[str] (default: None)

Prefix of saved file names.

backup_url : str | NoneOptional[str] (default: None)

URL to retrieve saved outputs from if not present on disk.

Returns

Model with loaded state dictionaries.

Examples

>>> model = ModelClass.load(save_path, adata) # use the name of the model class used to save
>>> model.get_....