利用surprise+自己的数据构建推荐系统
-
surprise介绍
surprise是一个python的推荐系统的库,里面包含数据下载、数据处理、模型(包括svd、svd++、slopeone、random rating base 等,具体可以去看官方文档地址)、评分等各种功能 -
实现
-
安装
1、在命令行 pip install surprise 2、命令行安装报错的话可以去[python库网站](https://www.lfd.uci.edu/~gohlke/pythonlibs/#qimage2ndarray)ctrl+f搜索surprise下载,然后在目录下shift+右键进入命令行 pip install ******.whl来安装 3、安装结束打开编辑器编辑一个python文件 `import surprise`并运行,不报错的话就完成了安装
-
数据
我们需要的数据有三行:user、item、ratings -
导入所需要的模块
import surprise
from surprise import Dataset,Reader
#Reader 将数据转化为surprise所需要的数据时需要这个