sccross.models.sccross.AnnDataset

class sccross.models.sccross.AnnDataset(adatas, data_configs, mode='train', getitem_size=1)[source]

Bases: Generic[torch.utils.data.dataset.T_co]

Dataset for anndata.AnnData objects with partial pairing support.

Parameters

  • *adatas – An arbitrary number of configured anndata.AnnData objects

  • data_configs (List[Mapping[str, Any]]) – Data configurations, one per dataset

  • mode (str) – Data mode, must be one of {"train", "eval"}

  • getitem_size (int) – Unitary fetch size for each __getitem__ call

Methods

accept_shuffle

Accept shuffling result

propose_shuffle

Propose shuffling using a given random seed

random_split

Randomly split the dataset into multiple subdatasets according to given fractions.

Attributes

adatas

Internal AnnData objects

data_configs

Data configuration for each dataset

logger