deep-rec 0.0.4.dev1

PyTorch implementation of Deep Factorization Machine Models

Factorization Machine models in PyTorch

This package provides a PyTorch implementation of Deep Factorization Machine ;odels and common datasets in Retail Recommendation.

Available Datasets

• Retail Case Study Data

Available Models

Model	Reference
DeepFM	H Guo, et al. DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, 2017.
xDeepFM	J Lian, et al. xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems, 2018.

Environment

conda env create -f environment_conda.yml
source activate environment_conda

Installation

pip install deep-rec

Example

python main.py --device cpu --epoch 2 

API Documentation

https://rixwew.github.io/deep-rec (en construcción)

Licence

MIT