cascade.data.SimpleDataModule

class cascade.data.SimpleDataModule(adata, batch_size, pin_memory, val_frac, random_state)

Bases: DataModule

Simple data module using SimpleDataset

Parameters:

• adata (AnnData) – Interventional dataset

• batch_size (int) – Batch size

• pin_memory (bool) – Whether to use pin memory

• val_frac (float) – Fraction of validation data

• random_state (RandomState | int | None) – Random state