tensorflow推荐系统
top-hat的Python项目详细描述
顶帽
tophat是一个基于因子分解的推荐引擎,使用 TensorFlow。
安装
从pypi安装:
pip install top-hat
在开发模式下从github安装主分支,运行:
git clone git@github.com:gilt/tophat.git
cd tophat
pip install -e .
注意,默认情况下,安装假定您已经安装了tensorflow。 但是,如果需要,可以在安装附加程序中包括TensorFlow的安装,如下所示: (选择一个适合您的)
# 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图像
提供了两个dockerfile:Dockerfile
和Dockerfile.gpu
。后一种gpu变体需要nvidia-docker。
测试
在tests/
中有一些最小的测试,它们都可以使用pytest
或python setup.py test
运行。
相关项目
tophat背后的最初动机是将LightFM和Spotlight导入tensorflow。
还有许多其他的amazing recommender systems out there --所以选择一个适合你的案子。