Please use pip install -r requirements.txt to setup the operating environment in python3.5.
Note that we use DeepCTR package and refer to the implementation of DSIN.
- Download Alimama Data: Ad Display/Click Data on Taobao.com
- Extract the files into the
data/raw_datadirectory - Follow the code of the data preprocessing in DSIN to preprocess data
Run python train_[model].py [model_type]
- Sum Pooling:
python train_base.py Base - MAF-S:
python train_base.py Base_All_Fields_Add - MAF-C:
python train_base.py Base_All_Fields_Concat - DIN:
python train_din.py DIN - DIN+:
python train_din.py All_Fields - DIEN:
python train_dien.py - DSIN:
python train_dsin.py - DotProduct:
python train_autoattention.py DotProduct - AutoAttention:
python train_autoattention.py AutoAttention_Prun 0.6