metadata
license: mit
library_name: pytorch tags: - recommender-systems - sasrec - movielens - pytorch
NextClick: SASRec Model trained on MovieLens 1M 🎬
This is the pre-trained PyTorch model for NextClick, a sequential recommendation engine.
Model Details
- Architecture: SASRec (Self-Attentive Sequential Recommendation).
- Embedding Size: 64.
- Max Sequence Length: 50.
- Dataset: MovieLens 1M (filtered for sessions > 5 interactions).
Usage (PyTorch)
import torch
# Load Architecture code first (See GitHub), then:
model.load_state_dict(torch.load("sasrec_model.pth"))
model.eval()
GitHub Repository
For the training code and architecture definition, visit: https://github.com/eatosin/NextClick-RecSys-Engine ```