Skip to main content

PyTorch implementation of Deep Factorization Machine Models

Project description

Factorization Machine models in PyTorch

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

Available Datasets

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

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

deep-rec-0.0.5.tar.gz (6.6 kB view hashes)

Uploaded Source

Built Distribution

deep_rec-0.0.5-py3-none-any.whl (8.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page