Skip to content

A PyTorch implementation of DeepFM for CTR prediction problem.

Notifications You must be signed in to change notification settings

blldd/DeepFM_with_PyTorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DeepFM_with_PyTorch

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

About

A PyTorch implementation of DeepFM for CTR prediction problem.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%