The data is available at Kaggle and in the cleaned, pre-processed data is available in the folder data.
To train a standard feed-forward NCM and compute counterfactual queries on the depression dataset, run all cells in depression.ipynb.
To train a standard ResNCM and compute counterfactual queries on the depression dataset, run all cells in depressionresnet.ipynb.
To train a standard feed-forward NCM and compute counterfactual queries on the exam scores dataset, run all cells in exam_scores.ipynb.
To train a standard ResNCM and compute counterfactual queries on the exam scores dataset, run all cells in exam_scores.ipynb.
The baselines and the data cleaning/pre-processing code can be found in the ipynb filed titled baselines.ipynb