pmixer/SASRec.pytorch

해결하는 문제

이 프로젝트는 사용자의 이전 상호작용 시퀀스를 기반으로 사용자가 다음에 상호작용할 아이템을 예측하기 위해 설계된 SASRec(Self-Attentive Sequential Recommendation) 모델의 PyTorch 구현을 제공합니다.

작동 방식

순차적인 사용자 행동을 모델링하기 위해 자기 주의(self-attention) 메커니즘을 구현합니다. 원래의 TensorFlow 구현을 PyTorch(v1.6)로 수정하여 위치 임베딩 사용 문제를 수정함으로써 과적합을 줄였습니다.

대상 사용자

PyTorch를 TensorFlow보다 선호하는 추천 시스템 개발자 및 연구자.

주요 특징

• 자기 주의 순차적 추천 논문의 PyTorch 기반 구현.
• ml-1m과 같은 데이터셋용 트레이닝 및 추론 스크립트 포함.
• LayerNorm을 추가하여 성능 지표(NDCG 및 HR)를 향상시켰습니다.