一、簡介

伯樂是用python中的pytorch實現的一個方法復現工具。用主流模型和主流數據來實現推薦系統中各種baseline

二、資源

github

https://github.com/RUCAIBox/RecBole/blob/master/README_CN.md

使用方法的入門博客

https://blog.csdn.net/Turinger_2000/article/details/111182852

三、使用其他數據（默認的ml-100k）完成測試

常規推薦只需要使用inner文件即可完成，不需要其他的輔助數據。

1.下載實驗數據(密碼e272)　　https://pan.baidu.com/share/init?surl=p51sWMgVFbAaHQmL4aD_-g

2.將下載的數據集中的.inter文件復制到模型根目錄中dataset中：

先在RecBole-master/dataset 新建一個與inter文件同名的文件夾，例如gowalla（數據名稱是gowalla.inter）

3.之后按照之前介紹的方法調用API接口進行測試例如

①新建一個python文件 如名為test_recbole.py

from recbole.quick_start import run_recbole
run_recbole()

②在命令行執行剛才新建的python文件，格式如下

python test_recbole.py --dataset=gowalla --model=BPR

總結

以上是生活随笔為你收集整理的推荐系统中对比实验工具--伯乐bole的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。