scvi.dataloaders.ConcatDataLoader

class scvi.dataloaders.ConcatDataLoader(adata, indices_list, shuffle=False, batch_size=128, data_and_attributes=None, drop_last=False, **data_loader_kwargs)[source]

DataLoader that supports a list of list of indices to load.

Parameters
adata : AnnDataAnnData

AnnData object that have been registered via setup_anndata().

indices_list : List[List[int]]List[List[int]]

List where each element is a list of indices in the adata to load

shuffle : boolbool (default: False)

Whether the data should be shuffled

batch_size : Optional[int]

minibatch size to load each iteration

data_and_attributes : dict | NoneOptional[dict] (default: None)

Dictionary with keys representing keys in data registry (adata.uns[“_scvi”]) and value equal to desired numpy loading type (later made into torch tensor). If None, defaults to all registered data.

data_loader_kwargs

Keyword arguments for DataLoader

Attributes

multiprocessing_context

Methods

check_worker_number_rationality()