cascade.data.PairedDataModule

class cascade.data.PairedDataModule(pri, sec, batch_size, pin_memory, val_frac, random_state)

Bases: DataModule

Paired data module using PairedDataset

Parameters:

• pri (AnnData) – Primary interventional dataset

• sec (AnnData) – Secondary 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