项目作者: chenxijun1029

项目描述 :
A PyTorch implementation of DeepFM for CTR prediction problem.
高级语言: Python
项目地址: git://github.com/chenxijun1029/DeepFM_with_PyTorch.git
创建时间: 2018-08-16T13:38:50Z
项目社区:https://github.com/chenxijun1029/DeepFM_with_PyTorch

开源协议:

下载


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.

    1. ./utils/dataPreprocess.py
  3. Train a model and predict.

    1. ./main.py

Output

Reference