File size: 840 Bytes
a3028a7 a269087 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | ---
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)
``` |