| --- |
| 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) |
| ```python |
| 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](https://github.com/eatosin/NextClick-RecSys-Engine) |
| ``` |
