Need help with DeepFM_with_PyTorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

chenxijun1029
246 Stars 83 Forks 19 Commits 8 Opened issues

Description

A PyTorch implementation of DeepFM for CTR prediction problem.

Services available

!
?

Need anything else?

Contributors list

# 224,010
Python
pytorch
ctr-pre...
mcmc
1 commit
# 419,073
Python
pytorch
ctr-pre...
1 commit

DeepFMwithPyTorch

A PyTorch implementation of DeepFM for CTR prediction problem.

Usage

  1. Download Criteo's Kaggle display advertising challenge dataset from here( if you have had it already, skip it ), and put it in ./data/raw/

  2. Generate a preprocessed dataset.

    ./utils/dataPreprocess.py
    
  3. 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.

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.