rectorch

pytorch-based framework for top-N recommendation. It includes several state-of-the-art top-N recommendation approaches implemented in pytorch

Build Status PyPi DOI Coverage Status docs version issues license

Included methods

The latest PyPi release contains the following methods.

Name Description Ref.
MultiDAE Denoising Autoencoder for Collaborative filtering with Multinomial prior [1]
MultiVAE Variational Autoencoder for Collaborative filtering with Multinomial prior [1]
CMultiVAE Conditioned Variational Autoencoder [2]
CFGAN Collaborative Filtering with Generative Adversarial Networks [3]
EASE Embarrassingly shallow autoencoder for sparse data [4]
ADMM_Slim ADMM SLIM: Sparse Recommendations for Many Users [5]
SVAE Sequential Variational Autoencoders for Collaborative Filtering [6]
RecVAE RecVAE: A New Variational Autoencoder for Top-N Recommendations with Implicit Feedback [7]

Now rectorch also includes some baseline methods.

Name Description Ref.
Random Random recommender
Popularity Popularity-based recommender
SLIM SLIM: Sparse Linear Methods for Top-N Recommender Systems [8]
CF-KOMD Boolean kernels for collaborative filtering in top-N item recommendation [9]

Getting started

Installation

rectorch is available on PyPi and it can be installed using pip

pip3 install rectorch

Requirements

If you install rectorch by cloning this repository make sure to install all the requirements.

pip3 install -r requirements.txt

Architecture

rectorch is composed of 9 main modules summarized in the following.

Name Scope
configuration Contains useful classes to manage the configuration files.
data Manages the reading, writing and loading of the data sets
evaluation Contains utility functions to evaluate recommendation engines.
metrics Contains the definition of the evaluation metrics.
models Includes the training algorithm for the implemented recommender systems.
nets Contains definitions of the neural newtork architectures used by the implemented approaches.
samplers Contains definitions of sampler classes useful when training neural network-based models.
utils Contains definitions of some utility functions.
validation Contains methods and classes for performing model selection.

Tutorials

We released a series of python notebooks with examples on how to train and evaluate recommendation methods using rectorch.

Documentation

The full documentation of the rectorch APIs is available at https://makgyver.github.io/rectorch/.

Known issues

The documentation has rendering issues on 4K display. To “fix” the problem zoom in ([Ctrl][+], [Cmd][+]) the page. Thanks for your patience, it will be fixed soon.

Testing

The easiest way to test rectorch is using pytest.

git clone https://github.com/makgyver/rectorch.git
cd rectorch/tests
pytest

You can also check the coverage using coverage. From the tests folder:

coverage run -m pytest  
coverage report -m

Dev branch

rectorch is developed using a test-driven approach. The master branch (i.e., the pypi release) is the up-to-date version of the framework where each module has been fully tested. However, new untested or under development features are available in the dev branch. The dev version of rectorch can be used by cloning the branch.

git clone -b dev https://github.com/makgyver/rectorch.git
cd rectorch
pip3 install -r requirements.txt

Work in progress

The following features/changes will be soon released:

  • Adding new baselines: BPR, WRMF, …
  • Adding new state-of-the-art models: NeuMF, …
  • Tutorials
  • Adding a set of methods/classes for performing series of experiments

Suggestions

This framework is constantly growing and the implemented methods are chosen on the basis of the need of our research activity. We plan to include as many state-of-the-art methods as soon as we can, but if you have any specific request feel free to contact us by opening an issue.

Citing this repo

If you are using rectorch in your work, please consider citing this repository.

@misc{rectorch,
    author = {Mirko Polato},
    title = {rectorch: pytorch-based framework for top-N recommendation},
    year = {2020},
    month = {sep},
    doi = {10.5281/zenodo.3841898},
    version = {0.9.0dev},
    publisher = {Zenodo},
    url = {https://doi.org/10.5281/zenodo.153841898991}
}

References

[1] Dawen Liang, Rahul G. Krishnan, Matthew D. Hoffman, and Tony Jebara. 2018. Variational Autoencoders for Collaborative Filtering. In Proceedings of the 2018 World Wide Web Conference (WWW ’18). International World Wide Web Conferences Steering Committee, Republic and Canton of Geneva, CHE, 689–698. DOI: https://doi.org/10.1145/3178876.3186150

[2] Tommaso Carraro, Mirko Polato and Fabio Aiolli. Conditioned Variational Autoencoder for top-N item recommendation, 2020. arXiv pre-print: https://arxiv.org/abs/2004.11141

[3] Dong-Kyu Chae, Jin-Soo Kang, Sang-Wook Kim, and Jung-Tae Lee. 2018. CFGAN: A Generic Collaborative Filtering Framework based on Generative Adversarial Networks. In Proceedings of the 27th ACM International Conference on Information and Knowledge Management (CIKM ’18). Association for Computing Machinery, New York, NY, USA, 137–146. DOI: https://doi.org/10.1145/3269206.3271743

[4] Harald Steck. 2019. Embarrassingly Shallow Autoencoders for Sparse Data. In The World Wide Web Conference (WWW ’19). Association for Computing Machinery, New York, NY, USA, 3251–3257. DOI: https://doi.org/10.1145/3308558.3313710

[5] Harald Steck, Maria Dimakopoulou, Nickolai Riabov, and Tony Jebara. 2020. ADMM SLIM: Sparse Recommendations for Many Users. In Proceedings of the 13th International Conference on Web Search and Data Mining (WSDM ’20). Association for Computing Machinery, New York, NY, USA, 555–563. DOI: https://doi.org/10.1145/3336191.3371774

[6] Noveen Sachdeva, Giuseppe Manco, Ettore Ritacco, and Vikram Pudi. 2019. Sequential Variational Autoencoders for Collaborative Filtering. In Proceedings of the Twelfth ACM International Conference on Web Search and Data Mining (WSDM ’19). Association for Computing Machinery, New York, NY, USA, 600–608. DOI: https://doi.org/10.1145/3289600.3291007

[7] Ilya Shenbin, Anton Alekseev, Elena Tutubalina, Valentin Malykh, and Sergey I. Nikolenko. 2020. RecVAE: A New Variational Autoencoder for Top-N Recommendations with Implicit Feedback. In Proceedings of the 13th International Conference on Web Search and Data Mining (WSDM ‘20). Association for Computing Machinery, New York, NY, USA, 528–536. DOI:https://doi.org/10.1145/3336191.3371831

[8] Mirko Polato and Fabio Aiolli. 2018. Boolean kernels for collaborative filtering in top-N item recommendation. Neurocomputing, Elsevier Science Ltd., Vol. 286, pp. 214-225, Oxford, UK. DOI: https://doi.org/10.1016/j.neucom.2018.01.057, ISSN: 0925-2312.

[9] X. Ning and George Karypis. 2011. SLIM: Sparse Linear Methods for Top-N Recommender Systems. In the Proceedings of the IEEE 11th International Conference on Data Mining, Vancouver, BC, 2011, pp. 497-506. DOI: https://doi.org/10.1109/ICDM.2011.134.