pmixer/SASRec.pytorch
PyTorch(1.6+) implementation of https://github.com/kang205/SASRec
解决的问题
本项目提供了基于用户先前交互序列预测用户下一步将交互的项目所需的SASRec（自注意力序列推荐）模型的PyTorch实现。
工作原理
通过实现自注意力机制来建模用户的顺序行为。该项目将原始的TensorFlow实现修改为使用PyTorch（v1.6），修复了位置嵌入使用中的问题，以减少过拟合。
适用人群
偏好使用PyTorch而非TensorFlow的推荐系统研究人员和开发者。
主要亮点
- 基于PyTorch的自注意力序列推荐论文实现。
- 包含用于ml-1m等数据集的训练和推理脚本。
- 更新了LayerNorm以提升性能指标（NDCG和HR）。
相关
- 项目
- 项目
- 项目
- 项目