RecBole

伯乐,一个统一、全面、高效的推荐系统代码库


  安  装   快速开始 arXiv
通用和可扩展的数据结构

我们设计了通用和可扩展的数据结构来支持各种推荐数据集统一化格式和使用。

全面的基准模型和数据集

我们实现了78个常用的推荐算法,并提供了28个推荐数据集的格式化副本。

大规模的标准评测

我们支持一系列被广泛认可的评估方式来测试和比较不同的推荐算法。





RecBole 是一个基于 PyTorch 实现的,面向研究者的,易于开发与复现的,统一、全面、高效的推荐系统代码库,它可以通过pip, Conda 和源文件三种途径安装,使用简单便捷。
详情见 【 安装RecBole 】



我们实现了78个推荐系统模型,包含常见的推荐系统类别,如: General Recommendation、 Sequential Recommendation、 Context-aware Recommendation、 Knowledge-based Recommendation, 能够满足基本的推荐系统科研需要。详情见【 模型列表 】



我们约定了一个统一、易用的数据文件格式,并已支持 28 个 benchmark dataset。用户可以选择使用我们的数据集预处理脚本,或直接下载已被处理好的数据集文件。
详情见【 数据集列表 】



RecBole面向研究者,免费提供给高校、师生、和任何推荐领域的爱好者。我们会提供定期的维护、更新。同时我们欢迎有意向的人员加入我们, 让我们一起构建更加美好的推荐系统开源社区!详情见【 关于 】





“世有伯乐,然后有千里马。千里马常有,而伯乐不常有。”——韩愈《马说》