/CATESelection

Sklearn-style implementations of model selection criteria for CATE estimation

Primary LanguagePython

CATESelection

Sklearn-style implementations of model selection criteria for CATE estimation.

This repo contains code to replicate the results presented in the ICML23 paper 'In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimation'.

Citing

If you use this software please cite the corresponding paper:

@inproceedings{curth2023search,
  title={In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimation},
  author={Curth, Alicia and van der Schaar, Mihaela},
  booktitle={International Conference on Machine Learning},
  year={2023},
  organization={PMLR}
}