Skip to content

Commit 73860dc

Browse files
committed
Fix model desc
1 parent addee02 commit 73860dc

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

README.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,13 @@ Recommendation(Click-Through Rate Prediction) Framework for Go, including:
1414

1515
### [Simple 2 layer MLP](./model/mlp/mlp.go)
1616

17+
GAUC on MovieLens 20M: **0.771258**
18+
1719
- [x] [Simple 2 layer MLP test on MovieLens](./example/movielens/feature_test.go)
1820
- [x] Dropout and L2 regularization
1921
- [x] Batch Normalization
2022

2123
### [YouTube DNN](./model/youtube/dnn.go)
22-
<img src="./art/youtube.png" alt="YouTube DNN" width="400"/>
2324

2425
GAUC on MovieLens 20M: **0.760381**
2526

@@ -28,13 +29,11 @@ GAUC on MovieLens 20M: **0.760381**
2829
- [ ] Batch Normalization
2930

3031
### [Deep Interest Network](./model/din/din.go)
31-
<img src="./art/din.png" alt="Deep Interest Network" width="400"/>
3232

3333
GAUC on MovieLens 20M: **0.790542**
3434

3535
- [x] [DIN test on MovieLens](./example/movielens/dinimpl_test.go)
36-
- [x] [Euclidean Distance based attention](model/activation.go)
37-
- [x] [Cosine Similarity based attention](model/activation.go)
36+
- [x] [Euclidean Distance](model/activation.go) and [Cosine Similarity](model/activation.go) based attention
3837
- [x] Dropout and L2 regularization
3938
- [ ] Batch Normalization
4039

0 commit comments

Comments
 (0)