EATosin commited on
Commit
a3028a7
·
verified ·
1 Parent(s): a8547ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -3
README.md CHANGED
@@ -1,3 +1,28 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ ---
5
+ library_name: pytorch
6
+ tags:
7
+ - recommender-systems
8
+ - sasrec
9
+ - movielens
10
+ - pytorch
11
+ ---
12
+
13
+ # NextClick: SASRec Model trained on MovieLens 1M 🎬
14
+
15
+ This is the pre-trained PyTorch model for **NextClick**, a sequential recommendation engine.
16
+
17
+ ### Model Details
18
+ * **Architecture:** SASRec (Self-Attentive Sequential Recommendation).
19
+ * **Embedding Size:** 64.
20
+ * **Max Sequence Length:** 50.
21
+ * **Dataset:** MovieLens 1M (filtered for sessions > 5 interactions).
22
+
23
+ ### Usage (PyTorch)
24
+ ```python
25
+ import torch
26
+ # Load Architecture code first (See GitHub), then:
27
+ model.load_state_dict(torch.load("sasrec_model.pth"))
28
+ model.eval()