Published February 6, 2020 | Version v1
Dataset Open

Synthetic Data Set for Uplift Modeling



This dataset is designed and simulated for evaluating uplift modeling and feature selection methods. The main feature of this dataset is that it generates features with various patterns associated with the outcome variable and the causal effect (or treatment effect). Thus it is suitable for evaluating feature importance and model interpretation for uplift modeling.

This dataset consists of 100 trials (replicates with different random seeds), each trial with 10,000 samples and 36 features. The outcome variable is binary, that makes this dataset for classification problem. The samples are equally split for control and treatment group (5,000 samples in each group in each trial).

The generated data has three types of features: (1) uplift features influencing the treatment effect on the conversion probability; (2) classification features affecting the conversion probability but independent of the treatment effect; and (3) irrelevant features that are independent of both conversion probability and the treatment effect. To model the relationship between uplift features and the treatment effect and classification features and outcome probability, we implement six types of association patterns in the data generation process: linear, quadratic, cubic, ReLU (Rectified Linear Unit), trigonometric function sine, and cosine.

In this data set, there are 36 features in total, including 10 classification features, 6 uplift features, and 20 irrelevant features.

Column names:

  • Trial ID: 'trial_id'
  • Experiment group label: 'treatment_group_key'
  • Outcome variable (classification label):  'conversion'
  • Feature names: ['x1_informative',
  • True underlying control conversion probability: 'control_conversion_prob'
  • True underlying treatment conversion probability: 'treatment1_conversion_prob'
  • True treatment effect:  'treatment1_true_effect'



Files (787.4 MB)

Name Size Download all
787.4 MB Preview Download