End-To-End Memory Networks
TensorFlow version == 1.4
* #https://arxiv.org/abs/1503.08895
* babi (tasks_1-20_v1-2)
* https://research.fb.com/downloads/babi/
* End to End Memory Network를 구현한 class
* babi dataset을 전처리 하는 코드.
* 전처리된 babi dataset을 학습하는 코드.
Jointly Learning
Model A: Embedding size = 50, Adjacent, Position Encoding, Temporal Encoding
Model B: Embedding size = 100, Adjacent, Position Encoding, Temporal Encoding, Relu
Task | Model A | Model B |
---|---|---|
1 | 0.998 | 0.999 |
2 | 0.828 | 0.756 |
3 | 0.7 | 0.788 |
4 | 0.991 | 0.998 |
5 | 0.935 | 0.983 |
6 | 0.983 | 1.0 |
7 | 0.888 | 0.951 |
8 | 0.917 | 0.966 |
9 | 0.989 | 1.0 |
10 | 0.977 | 0.999 |
11 | 0.999 | 1.0 |
12 | 0.973 | 0.999 |
13 | 0.999 | 1.0 |
14 | 1.0 | 0.996 |
15 | 1.0 | 1.0 |
16 | 0.449 | 0.46 |
17 | 0.614 | 0.602 |
18 | 0.92 | 0.917 |
19 | 0.244 | 0.184 |
20 | 1.0 | 1.0 |
total | 0.8702 | 0.8799 |