scvi.dataloaders.AnnTorchDataset

class scvi.dataloaders.AnnTorchDataset(adata, getitem_tensors=None)[source]

Extension of torch dataset to get tensors from anndata.

Attributes

registered_keys

Returns the keys of the mappings in scvi data registry.

Methods

get_data(scvi_data_key)

setup_data_attr()

Sets data attribute.

setup_getitem()

Sets up the __getitem__ function used by Pytorch.