cascade.data

Dataset processing utilities and data loaders

Functions

aggregate_obs	Aggregate obs in a given dataset by certain categories
configure_dataset	Configure dataset for model training and inferences
encode_regime	Encode intervention regime
filter_unobserved_targets	Filter cells where the intervention targets are not observed (missing from var_names)
get_all_targets	Get the union of all intervention targets in a give dataset
get_configuration	Retrieve the configuration by configure_dataset()
neighbor_impute	Impute data by aggregating nearest neighbors
simple_design	Perform simple intervention design by directly comparing the outcome of seen interventions with the target

Classes

DataModule	Abstract data module
DynamicPairedDataModule	Dynamic paired data module using DynamicPairedDataset
DynamicPairedDataset	A pair of interventional datasets that are not paired but fetches randomly paired cells on-the-fly
PairedDataModule	Paired data module using PairedDataset
PairedDataset	A pair of interventional datasets that are paired cell-by-cell
SimpleDataModule	Simple data module using SimpleDataset
SimpleDataset	A single interventional dataset
Targets	Intervention targets