arctanln2/DeepFM_with_PyTorch

A PyTorch implementation of DeepFM for CTR prediction problem.

Python

DeepFM_with_PyTorch

A PyTorch implementation of DeepFM for CTR prediction problem.

Usage

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.
```
 ./utils/dataPreprocess.py
```
Train a model and predict.
```
 ./main.py
```

Output

Reference

https://github.com/nzc/dnn_ctr.
https://github.com/PaddlePaddle/models/tree/develop/deep_fm.
DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, Huifeng Guo, Ruiming Tang, Yunming Yey, Zhenguo Li, Xiuqiang He.