Downloads · 30 days
0
sasrec-pytorch
MongrelIntruder/sasrec-pytorch
sasrec-pytorch is a machine learning model from MongrelIntruder. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
PyTorch reimplementation of SASRec (Kang & McAuley, 2018), trained as part of a thesis study
Access
Public
Updated Aug 28, 2026
Repo size
1.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin858 KB · 98%
At a glance
- Access
- Public
- Created
- Aug 16, 2026
- Updated
- Aug 28, 2026
- SHA
- 377cf57f
Model card
From the Hugging Face model README
SASRec (ml-1m, norm_first=False)
PyTorch reimplementation of SASRec (Kang & McAuley, 2018), trained as part of a thesis study
Best validation/test scores seen during training:
- valid: NDCG@10 0.6253, HR@10 0.8503
- test: NDCG@10 0.5970, HR@10 0.8260
Files
pytorch_model.binmodel.pyutils.pyargs.json
Loading
import torch, json
from types import SimpleNamespace
from model import SASRec
from utils import build_index, data_partition
args = SimpleNamespace(**json.load(open("args.json")))
u2i_index, i2u_index = build_index(args.dataset)
dataset = data_partition(args.dataset)
usernum, itemnum = dataset[3], dataset[4]
model = SASRec(usernum, itemnum, args)
model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
model.eval()
- Created
- Aug 16, 2026
- Updated
- Aug 28, 2026