#16 · 主分类： 推荐系统
spotlight
deep-learning
learning-to-rank
machine-learning
matrix-factorization
python
pytorch
recommender-system
基于PyTorch的深度推荐模型。
项目最后更新：12/21/22
GitHub Stars
3.0K
Forks数量
418
贡献者数量
11
许可证
MIT
收录理由
Spotlight 是一个基于 PyTorch 的推荐模型库，它把两类常见的推荐思路统一在同一套接口里：一类是经典的矩阵分解模型，用于评分预测和排序；另一类是序列模型，用于预测用户下一个可能交互的物品。序列模型部分提供了池化、LSTM 和因果卷积等多种实现，而且所有模型共享相同的数据结构和评估工具，这意味着你可以在不同方法之间切换，而无需重写整个数据处理流程。库中还自带了 MovieLens 数据集的加载器和合成序列生成器，方便快速验证想法。整体设计比较模块化，如果你想尝试自定义损失函数或表示方式，扩展起来并不困难。需要注意的是，这个项目自 2022 年底以来就没有再更新过，因此更适合作为原型开发的起点，而不是直接部署到生产环境。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。