#6 · 主分类: 推荐系统

RecBole

collaborative-filtering ctr-prediction deep-learning graph-neural-networks knowledge-graph pytorch recommendation-system recommendations recommender recommender-systems sequential-recommendation

一个统一、全面且高效的推荐库

项目最后更新:02/24/25

GitHub Stars

4.5K

Forks数量

748

贡献者数量

79

许可证

MIT

收录理由

做推荐系统的研究人员和工程师,拿 RecBole 当起点几乎挑不出更省事的选项。库里有 94 个常见推荐算法的 PyTorch 实现,覆盖通用协同过滤、序列推荐、上下文感知和基于知识的模型,还附带 44 个基准数据集的预处理版本。所有东西共用同一套数据格式和训练流程,换模型、比基线不用改代码,也不用头疼不同数据集格式对不上。数据预处理脚本也集成在管线里,原始日志很快就能变成可复现的实验。要是追新方向,配套的 RecBole2.0 工具包把能力延伸到去偏、公平性这类前沿课题上。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目