Recommendation system in TensorFlow
Project description
Tophat
Tophat is a factorization-based recommendation engine built using TensorFlow.
Installation
Installing from PyPi:
pip install top-hat
Installing the master branch from github in development mode, run:
git clone git@github.com:gilt/tophat.git
cd tophat
pip install -e .
Note that by default, installation assumes you already have TensorFlow installed. However, if you need, you can include the installation of TensorFlow in the setup extras as following: (choose the one that's right for you)
# CPU pypi
pip install top-hat[tf]
# GPU pypi
pip install top-hat[tf_gpu]
# CPU local dev
pip install -e .[tf]
# GPU local dev
pip install -e .[tf_gpu]
Docker Images
There are two provided Dockerfiles: Dockerfile
and Dockerfile.gpu
. The latter gpu variant requires nvidia-docker.
Tests
There are some minimal tests in tests/
which can all be run using pytest
or python setup.py test
.
Related Projects
The initial motivation behind tophat was to port over LightFM and Spotlight into TensorFlow.
There also are many other amazing recommender systems out there -- so choose the one that is right for your case.
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
Built Distribution
Hashes for top_hat-0.0.1-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0641bb19fa0de0eefd1e342b700fd4377a5abbd7fb09615c883cae8bf043bbac |
|
MD5 | 385ce465198a46e94785b32388d7cb9f |
|
BLAKE2b-256 | 063812842fec8c31fe4dad0b4031d3506bb1ed366402c94ce81718dc5c747a80 |