scvi.dataloaders.AnnDataLoader#
- class scvi.dataloaders.AnnDataLoader(adata_manager, shuffle=False, indices=None, batch_size=128, data_and_attributes=None, drop_last=False, iter_ndarray=False, **data_loader_kwargs)[source]#
DataLoader for loading tensors from AnnData objects.
- Parameters:
- adata_manager :
AnnDataManager
AnnDataManager
object with a registered AnnData object.- shuffle
Whether the data should be shuffled
- indices
The indices of the observations in the adata to load
- batch_size : Optional[int]
minibatch size to load each iteration
- data_and_attributes :
dict
|None
Optional
[dict
] (default:None
) Dictionary with keys representing keys in data registry (
adata_manager.data_registry
) and value equal to desired numpy loading type (later made into torch tensor). IfNone
, defaults to all registered data.- data_loader_kwargs
Keyword arguments for
DataLoader
- iter_ndarray :
bool
(default:False
) Whether to iterate over numpy arrays instead of torch tensors
- adata_manager :
Attributes table#
Methods table#
Attributes#
multiprocessing_context#
- AnnDataLoader.multiprocessing_context#
dataset#
batch_size#
num_workers#
pin_memory#
drop_last#
timeout#
sampler#
prefetch_factor#
Methods#
check_worker_number_rationality#
- AnnDataLoader.check_worker_number_rationality()#