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