rockyzhengwu / AITM-torch

Pytorch implementation of Adaptive Information Transfer Multi-task (AITM)

AITM-torch

• Official Tensorflow implementation
• Paper

Dataset

Download and preprocess dataset use script in official Tensorflow implementation

python process_public_dataset.py

Train && Test

mkdir out
python train.py

AUC performance

Test AUC on Alibaba Click and Conversion Prediction dataset

Test Resutt: click AUC: 0.6189267022220789 conversion AUC:0.6544229866061039