Description:

deeprec 0.0.5

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

License

For personal and professional use. You cannot resell or redistribute these repositories in their original state.

Share

• Share on Facebook
• Share on Twitter