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.
Filename, size | File type | Python version | Upload date | Hashes |
---|---|---|---|---|
Filename, size top_hat-0.0.2-py2.py3-none-any.whl (53.3 kB) | File type Wheel | Python version py2.py3 | Upload date | Hashes View |
Filename, size top-hat-0.0.2.tar.gz (42.7 kB) | File type Source | Python version None | Upload date | Hashes View |
Hashes for top_hat-0.0.2-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 39a167c4a30e6d96ce788f0e8ce91e63097c1f8662b070c88f3e272d6916eb6e |
|
MD5 | d88582f6c63bb18ec54b6c4a8e6a771c |
|
BLAKE2-256 | 02d74e2de3c83a7d2723f28d1b2d5a241878540707bc73203bbde28ccefcbb12 |