DaisyRec:推荐系统基准的多维公平比较工具

Posted 2021-05-01 专知

【导读】DaisyRec 是一个Python包，用于物体得分预测与排序。DAISY起源于Multi-Dimension fAIrly compArIson for recommender SYstem.

Github链接：

https://github.com/AmazingDD/daisyRec

模块

整个工具包括数据加载、推荐算法、评估算法三个模块。

安装

安装所需的依赖包：

pip install -r requirement.txt

使用以下语句生成.so(MacOS) 与.pyd （Windows）文件

python setup.py build_ext --inplace

DaisyRec主要处理排序问题，并将推荐问题分为点对问题和成对问题，以构造不同的损失函数，如BPR，Top-1，Hinge 和交叉熵。下图概括了已实现的所有算法：

使用方法

运行main.py 并指定参数，可以指定不同的排序方法，算法，以及损失函数。

python main.py --problem_type=pair --algo_name=mf --loss_type=BPR --num_ng=2

专知，专业可信的人工智能知识分发，让认知协作更快更好！欢迎注册登录专知www.zhuanzhi.ai，获取5000+AI主题干货知识资料！

欢迎微信扫一扫加入专知人工智能知识星球群，获取最新AI专业干货知识教程资料和与专家交流咨询！

点击“ 阅读原文 ”，了解使用 专知 ，查看获取5000+AI主题知识资源