spVIPES.dataloaders._concat_dataloader.ConcatDataLoader

spVIPES.dataloaders._concat_dataloader.ConcatDataLoader#

class spVIPES.dataloaders._concat_dataloader.ConcatDataLoader#

Bases: DataLoader

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

Parameters:

  • adata_manager (AnnDataManager) – AnnDataManager object that has been created via setup_anndata.

  • indices_list (list[list[int]]) – List where each element is a list of indices in the adata to load

  • shuffle (bool (default: True)) – Whether the data should be shuffled

  • use_labels (bool (default: False)) – Whether to use labels for sampling

  • batch_size (int (default: 128)) – minibatch size to load each iteration

  • data_and_attributes (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). If None, defaults to all registered data.

  • data_loader_kwargs – Keyword arguments for DataLoader

__init__(adata_manager, indices_list, shuffle=True, use_labels=False, batch_size=128, data_and_attributes=None, drop_last=False, **data_loader_kwargs)#
static __new__(cls, *args, **kwargs)#
Return type:

Any

Contents