Skip to content

Commit b0ceea8

Browse files
committed
Update readme
1 parent 56f164d commit b0ceea8

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

README.md

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,22 @@
44

55
Deep Learning(Item2vec Embedding + MLP) based Feature-Engineering & Training & Predict all in one Recommendation System that can run on small server or edge device.
66

7+
# Models implemented
8+
9+
- [x] [Simple 2 layer MLP](./nn/neural_network/multilayer_perceptron.go)
10+
- [x] [Simple 2 layer MLP test on MovieLens](./example/movielens/feature_test.go)
11+
- [x] Dropout and L2 regularization
12+
- [x] Batch Normalization
13+
- [x] [YouTube DNN](./model/din/simplemlp.go)
14+
- [x] [YouTube DNN test on MovieLens](./example/movielens/mlpimpl_test.go)
15+
- [x] Dropout and L2 regularization
16+
- [ ] Batch Normalization
17+
- [x] [DeepInterestNetwork](./model/din/din.go)
18+
- [x] [DIN test on MovieLens](./example/movielens/dinimpl_test.go)
19+
- [x] Euclidean Distance based attention
20+
- [x] Dropout and L2 regularization
21+
- [ ] Batch Normalization
22+
723
# Demo
824

925
You can run the MovieLens training and predict demo by:
@@ -161,6 +177,7 @@ To make this project work, quite a lot of code are copied and modified from the
161177
162178
# Papers related
163179
180+
- [YouTube DNN](https://static.googleusercontent.com/media/research.google.com/en//pubs/archive/45530.pdf)
164181
- [Deep Interest Network for Click-Through Rate Prediction](https://arxiv.org/abs/1706.06978)
165182
- [Document Embedding with Paragraph Vectors](https://arxiv.org/abs/1507.07998)
166183
- [EdgeRec: Recommender System on Edge in Mobile Taobao](https://arxiv.org/abs/2005.08416) // not very identical implementation

0 commit comments

Comments
 (0)