catenets.datasets package

load(dataset: str, *args: Any, **kwargs: Any) → Tuple

Input:

dataset: the name of the dataset to load

Outputs:

• Train_X, Test_X: Train and Test features

• Train_Y: Observable outcomes

• Train_T: Assigned treatment

• Test_Y: Potential outcomes.

Submodules

• catenets.datasets.dataset_acic2016 module
• catenets.datasets.dataset_ihdp module
• catenets.datasets.dataset_twins module
• catenets.datasets.network module