A Python Library for Recommender System
Project description
a Python library for recommender system, which is hyper-modular, extensively tested and easy to extend. Reclib makes it easy to design and evaluate deep learning models for recommender system, along with the infrastructure to easily run them in the cloud or on your laptop.
The CLI part is under construction for now.
Package Overview
reclib | an Python library for recommender system |
---|---|
reclib.commands | functionality for a CLI and web service |
reclib.data | a data processing module for loading and encoding datasets for representation |
reclib.models | a collection of state-of-the-art models |
reclib.modules | a collection of PyTorch modules for use with recommender system |
reclib.nn | tensor utility functions, such as initializers and activation functions |
reclib.service | a web server to that can serve demos for your models |
reclib.training | functionality for training models |
Installation
reclib requires Python 3.7.1 or later. The preferred way to install reclib is via pip
. Just run pip install reclib
in your Python environment and you're good to go!
If you need pointers on setting up an appropriate Python environment or would like to install reclib using a different method, see below.
Windows is currently not officially supported, although we try to fix issues when they are easily addressed.
Installing via pip
Setting up a virtual environment
Conda can be used set up a virtual environment with the version of Python required for reclib. If you already have a Python 3.7 or 3.7 environment you want to use, you can skip to the 'installing via pip' section.
-
Create a Conda environment with Python 3.7
conda create -n reclib python=3.7
-
Activate the Conda environment. You will need to activate the Conda environment in each terminal in which you want to use reclib.
source activate reclib
Installing the library and dependencies
Installing the library and dependencies is simple using pip
.
pip install reclib
That's it! You're now ready to build and train reclib models.
reclib installs a script when you install the python package, meaning you can run reclib commands just by typing reclib
into a terminal.
You can now test your installation with reclib test-install
.
pip
currently installs Pytorch for CUDA 9 only (or no GPU). If you require an older version,
please visit https://pytorch.org/ and install the relevant pytorch binary.
Features
- Pythonic
- Easy to use
- State-of-the-art
Models
Please refer to the documents
MovieLens 1M
Model | MAP | nDCG@k | Precision@k | Recall@k | RMSE | MAE | R2 | Auc | Explained Variance |
---|---|---|---|---|---|---|---|---|---|
XDeepFM | 0.800158 | ||||||||
AutomaticFeatureInteraction | 0.7928 | ||||||||
AttentionalFactorizationMachine | 0.77824 | ||||||||
DeepCrossNetwork | 0.7928 | ||||||||
DeepFactorizationMachine | 0.7928 | ||||||||
FieldAwareFactorizationMachine | 0.7928 | ||||||||
AutomaticFeatureInteraction | 0.7928 | ||||||||
FactorizationMachine | 0.7928 | ||||||||
FieldAwareNeuralFactorizationMachine | 0.7928 | ||||||||
FactorizationSupportedNeuralNetwork | 0.7928 |
Issues
Everyone is welcome to file issues with either feature requests, bug reports, or general questions. As a small team with only one person, we may ask for contributions if a prompt fix doesn't fit into our roadmap. We allow users a two week window to follow up on questions, after which we will close issues. They can be re-opened if there is further discussion.
Contributions
If you would like to contribute a larger feature, we recommend first creating an issue with a proposed design for discussion. This will prevent you from spending significant time on an implementation which has a technical limitation someone could have pointed out early on. Small contributions can be made directly in a pull request.
Pull requests (PRs) must have one approving review and no requested changes before they are merged.
Licence
Apache 2.0
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.