scglue.models.data.Dataset
- class scglue.models.data.Dataset(getitem_size=1)[source]
Bases:
Dataset
Abstract dataset interface extending that of
torch.utils.data.Dataset
- Parameters
getitem_size (
int
) – Unitary fetch size for each __getitem__ call
Methods
Accept shuffling result
Clean up multi-process resources used in custom shuffling
Prepare dataset for custom shuffling
Propose shuffling using a given random seed
Custom shuffling
Background shuffle worker
Attributes
has_workers
Whether background shuffling workers have been registered
logger