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）。
相關
- 專案
- 專案
- 專案
- 專案