This is the official repository of the paper CounterNet: End-to-End Training of Counterfactual Aware Predictions. The purpose of the repository is only for research and reproduction of the paper's results. The audience should not expect to use the code directly in the deployed environemnt.
The repository is built based on nbdev. I highly recommend you to check out nbdev if you enjoy writing code with Jupyter Notebook as I do. Further, this project primarily leverages Pytorch
and Pytorch Lightning
for implementations of deep learning models. To install all the dependencies, you should run:
pip install -e .
Note:
pip install
will only install cpu-version ofpytorch
. If you want to use GPU-version ofpytorch
, please follow pytorch's official instruction.- As
Pytorch Lightning
's API changes rapidly, it is not guaranteed that the code is compatible with other versions of Lightning (except the version that specifiedsettings.ini
).
nbdev_build_lib
nbdev_update_lib
nbdev_clean_nbs
nbdev_test_nbs
tensorboard --logdir log --bind_all