Code for the paper:

Yinjun Wu, Mayank Keoliya, Kan Chen, Neelay Velingker, Ziyang Li, Emily J Getzen, Qi Long, Mayur Naik, Ravi B Parikh, Eric Wong, "DISCRET: Synthesizing Faithful Explanations For Treatment Effect Estimation" (ICML 2024)

Installing prerequisites

pip install -r requirements.txt

Run experiments

For experiments on tabular dataset, go to "treatment_prediction/tabular" folder. For experiments on NLP dataset, go to "treatment_prediction/nlp" folder. For experiments on image datasets, go to "treatment_prediction/image" folder

Citing

If you find our work useful in your research, please consider citing:

@inproceedings{wu2024discret,
	title={DISCRET: Synthesizing Faithful Explanations For Treatment Effect Estimation},
	author={Yinjun Wu, Mayank Keoliya, Kan Chen, Neelay Velingker, Ziyang Li, Emily J Getzen, Qi Long, Mayur Naik, Ravi B Parikh, Eric Wong}
  	booktitle={International Conference on Machine Learning (ICML)},
  	year={2024}
}#