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.txt
and run theUM-experiment.ipynb
notebook.
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.