cascade.data.DynamicPairedDataModule

class cascade.data.DynamicPairedDataModule(pri, sec, stratify, batch_size, pin_memory, val_frac, random_state)[source]

Bases: DataModule

Dynamic paired data module using DynamicPairedDataset

Parameters:

  • pri (AnnData) – Primary interventional dataset

  • sec (AnnData) – Secondary interventional dataset

  • stratify (str | None) – Column in obs used for stratification

  • 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