guotong1988 / run_criteo_dataset

Get AUC 0.809 at Criteo dataset by MLP
4 stars 1 forks source link
click-through-rate-prediction ctr-prediction deep-learning recommendation-system tensorflow

Run Criteo Dataset

Get AUC 0.809 at Criteo dataset by MLP

Requirements

python3

tensorflow 1.14 or 1.15

Reproduce

step0, download dataset from link1 link2 link3 (the total file is about 11G)

step1, run data/preprocess.py

step2, run train.py

More Info

https://paperswithcode.com/sota/click-through-rate-prediction-on-criteo