This repo contains the code for Could X cause Y? A causality study (CMU 10708 Project 2022).
Executing this code requires Python 3.6 along with the following packages:
- pandas (tested with version 10.1)
- sklearn (tested with version 0.13)
- numpy (tested with version 1.6.2)
- scipy (tested with version 0.10.)
- Pytorch (tested with version 1.0.0)
To run the code,
- Download the data, and run
python data2img.py
for data visualization. - Modify SETTINGS.json to point to the training and validation data on your system, as well as a place to save the trained model and a place to save the submission
- Train the model by running
python train.py
- Make predictions on the validation set by running
python predict.py
- Run
python score.py
for evaluation
To train and test the CNN, please refer to train_CNN.ipynb
.