DeepFM_with_PyTorch

by chenxijun1029

chenxijun1029 / DeepFM_with_PyTorch

A PyTorch implementation of DeepFM for CTR prediction problem.

165 Stars 63 Forks Last release: Not found 19 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

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.