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
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
Release history Release notifications | RSS feed
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)