sjchasel / DeepFM_with_PyTorch

A PyTorch implementation of DeepFM for CTR prediction problem.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

ezoic increase your site revenue

About

A PyTorch implementation of DeepFM for CTR prediction problem.


Languages

Language:Python 100.0%