RecBole

项目介绍：RecBole（伯乐）

RecBole 是一个基于 Python 和 PyTorch 的统一、综合且高效的推荐算法开发与复现框架，专为学术研究而设计。该项目的名称源自于韩愈的《马说》，意在强调伯乐对千里马的重要性。在大数据时代，RecBole 站在了推进科技发展的前沿，通过提供丰富的推荐系统模型与数据集，帮助研究者快速实现算法测试及新模型的开发。

RecBole 的主要功能

统一且可扩展的数据结构

RecBole 设计了统一且可扩展的数据结构，使得不同推荐系统数据集在格式与使用上实现统一。这一特点使得研究人员在开发过程中无需纠结于数据格式转换，专注于算法的优化与创新。

综合性的标准模型与数据集

RecBole 实现了 91 种常用的推荐算法，涵盖了通用推荐、序列推荐、上下文感知推荐以及基于知识的推荐四大类。此外，RecBole 提供了 43 个标准推荐数据集，用户可以利用团队提供的脚本来处理原始数据，或直接下载已经处理好的数据集，方便快捷。

高效的 GPU 加速执行

为了提高执行效率，RecBole 针对 GPU 环境进行了多项优化，使得模型在训练时可以高效利用 GPU 资源。在处理大规模数据集时，这一点尤为重要，能够显著缩短训练时间。

标准化的评估协议

RecBole 支持多种广泛应用的评估协议，帮助研究人员在测试和比较推荐算法时，能够得到可靠且一致的结果。这些评估协议涵盖了从简单的指标计算到复杂的实验设置，适用于不同的实验需求。

最新版本更新

RecBole 在最新的更新中着重提升了其易用性和用户友好性，使其更加适合研究人员作为推荐系统的基准库使用。在更新中，项目团队制定了详细的开发指南，改善了文档的组织结构，并增加了常见问题解答，提升了用户体验。

安装指南

RecBole 兼容 Linux、Windows 10 和 macOS X 操作系统，要求 Python 版本不低于 3.7，torch 版本不低于 1.7.0。以下是安装方法：

• 使用 conda 安装：

  conda install -c aibox recbole

• 使用 pip 安装：

  pip install recbole

• 从源代码安装：

  git clone https://github.com/RUCAIBox/RecBole.git && cd RecBole
  pip install -e . --verbose

快速入门

开发者可以使用 RecBole 提供的脚本进行快速体验，运行以下命令即可测试 BPR 模型在 ml-100k 数据集上的表现：

python run_recbole.py

运行后，用户可以观察到训练过程的详细信息以及模型评估结果，这为进一步的开发与优化提供了良好的起点。

开源贡献与合作

RecBole 项目拥有多个相关的 GitHub 项目，为数据处理、模型开发、算法训练和科学评估提供了一站式解决方案。项目欢迎各种形式的贡献，包括 Bug 修复、新功能开发，以及项目的扩展。

引用

如果 RecBole 为您的研究或开发提供了帮助，请参考以下论文进行引用：

• RecBole 1.0
• RecBole 2.0
• RecBole 1.2.0

通过这些论文，研究人员可以深入了解 RecBole 的开发背景及其在推荐系统领域的贡献。

感谢与支持

本项目得到了中国国家自然科学基金项目（No. 61832017）的支持，感谢所有为 RecBole 项目做出贡献的开发者和研究人员。项目使用 MIT 许可协议，所有数据和代码仅限用于学术用途。