遇见数据集

Synthetic Data Set for Uplift Modeling

收藏
Zenodo2020-07-30 更新2026-05-25 收录
数据链接:
官方服务:

资源简介:

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',<br> 'x2_informative',<br> 'x3_informative',<br> 'x4_informative',<br> 'x5_informative',<br> 'x6_informative',<br> 'x7_informative',<br> 'x8_informative',<br> 'x9_informative',<br> 'x10_informative',<br> 'x11_irrelevant',<br> 'x12_irrelevant',<br> 'x13_irrelevant',<br> 'x14_irrelevant',<br> 'x15_irrelevant',<br> 'x16_irrelevant',<br> 'x17_irrelevant',<br> 'x18_irrelevant',<br> 'x19_irrelevant',<br> 'x20_irrelevant',<br> 'x21_irrelevant',<br> 'x22_irrelevant',<br> 'x23_irrelevant',<br> 'x24_irrelevant',<br> 'x25_irrelevant',<br> 'x26_irrelevant',<br> 'x27_irrelevant',<br> 'x28_irrelevant',<br> 'x29_irrelevant',<br> 'x30_irrelevant',<br> 'x31_uplift_increase',<br> 'x32_uplift_increase',<br> 'x33_uplift_increase',<br> 'x34_uplift_increase',<br> 'x35_uplift_increase',<br> 'x36_uplift_increase'] True underlying control conversion probability: 'control_conversion_prob' True underlying treatment conversion probability: 'treatment1_conversion_prob' True treatment effect: 'treatment1_true_effect'

提供机构:
Zenodo
创建时间:
2020-02-07
二维码
社区交流群
二维码
科研交流群
商业服务