large-scale-ITE-UM-benchmark
This repository contains code and data to reproduce the results of the paper "A Large Scale Benchmark for Individual Treatment Effect Prediction and Uplift Modeling", submitted to NeurIPS 2021 Conference, Datasets and Benchmarks Track on OpenReview.
CRITEO-UPLIFTv2 dataset
- Webpage: https://ailab.criteo.com/criteo-uplift-prediction-dataset/
- Direct link to the data: http://go.criteo.net/criteo-research-uplift-v2.1.csv.gz
- To reproduce the results of Table 3 and Figure 3, please install the required packages from
requirements.txtand run theUM-experiment.ipynbnotebook.
CRITEO-ITE dataset
To generate the data and reproduce the results of Table 4, please install the required packages from requirements.txt and run the ITE-experiment.ipynb notebook.