A PyTorch implementation of DeepFM for CTR prediction problem.
Download Criteo's Kaggle display advertising challenge dataset from here( if you have had it already, skip it ), and put it in ./data/raw/
Generate a preprocessed dataset.
Train a model and predict.
DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, Huifeng Guo, Ruiming Tang, Yunming Yey, Zhenguo Li, Xiuqiang He.