Skip to content

Commit 9c7e3ac

Browse files
committed
add images
1 parent 73143a3 commit 9c7e3ac

File tree

6 files changed

+13
-9
lines changed

6 files changed

+13
-9
lines changed

README.MD

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# GRU
2+
Implemention for Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling. [Read paper](https://arxiv.org/pdf/1412.3555.pdf)
23

34
Library Logo <--- **FIXME**
45

@@ -11,9 +12,12 @@ Compare and evaluate recurrent neural networks (RNN) with different types of rec
1112
- Long short-term memory (LSTM)
1213
- Tanh
1314

14-
Slide about your project (if it's available) <--- **FIXME**
15+
Slide about our project [vietnamese](https://docs.google.com/document/d/1GHZIDAyfCL2O-PVPVyQJdcQ8FxdkPGA4-AQd3cCYoMQ/edit?usp=sharing)
1516

16-
Architecture Image <--- **FIXME**
17+
Architecture
18+
<p align="center">
19+
<img src='./assets/architecture.png' width="90%" class="center">
20+
</p>
1721

1822
Advisors:
1923
- Github: [bangoc123](https://github.com/bangoc123)
@@ -187,22 +191,22 @@ Epoch 10/10
187191
188192
```
189193

190-
### 4. Comments about these results
194+
### 4. Results
191195

192-
- Base on the results of Training with 3 kinds of model, we can figure out that there are differences about the Accuaracy between them
196+
- Base on the results of Training with 3 kinds of model, we can figure out that there are differences about the Accuracy between them
193197
+ Both GRU and LSTM have high efficiency with the highest val_accuracy is 0.8936 and 0.8795 , respectively.
194198
+ Tanh is the opposite, the val_accuracy is really low with just around 0.54
195199
- The results clearly demonstrated the superiority of the gated units; both the LSTM unit and GRU, over the traditional tanh unit.
196200
- However, we could not make concrete conclusion on which of the two gating units was better.
197201

198-
## VI. Running Test
202+
<!-- ## VI. Running Test
199203
200204
When you want to modify the model, you need to run the test to make sure your change does not affect the whole system.
201205
202206
In the `./folder-name` **(FIXME)** folder please run:
203207
204208
```bash
205209
pytest
206-
```
210+
``` -->
207211

208212

assets/architecture.png

120 KB
Loading

assets/gru.png

183 KB
Loading

assets/lstm.png

252 KB
Loading

predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import os
22
from argparse import ArgumentParser
33
import tensorflow as tf
4-
from model.gru_rnn import GRU_RNN
54
from data import Dataset
65
import pandas as pd
76
import numpy as np
7+
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
88

99
if __name__ == "__main__":
1010
parser = ArgumentParser()

train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,6 @@
123123
model.save(f"{args.model_folder}/{args.model}.h5py")
124124

125125
# Do Prediction
126-
print('==============Evaluate=============')
127-
model.evaluate(val_ds, batch_size=128)
126+
# print('==============Evaluate=============')
127+
# model.evaluate(val_ds, batch_size=128)
128128

0 commit comments

Comments
 (0)