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 ```

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support