Metadata-Version: 2.1
Name: deep-rec
Version: 0.0.0
Summary: PyTorch implementation of Deep Factorization Machine Models
Home-page: https://github.com/jaisenbe58r/deep-rec
Author: Jaime Sendra Berenguer
Author-email: jaimesendraberenguer@gmail.com
License: UNKNOWN
Platform: UNKNOWN
Description-Content-Type: text/markdown

# 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](https://www.kaggle.com/darpan25bajaj/retail-case-study-data/download)


## Available Models

| Model | Reference |
|-------|-----------|
| DeepFM | [H Guo, et al. DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, 2017.](https://arxiv.org/abs/1703.04247) |
| xDeepFM | [J Lian, et al. xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems, 2018.](https://arxiv.org/abs/1803.05170) |


## Environment


    conda env create -f environment_conda.yml
    source activate environment_conda


## Installation

    pip install rec-deep


## API Documentation

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


## Licence

MIT

