Update README.md
Browse files
README.md
CHANGED
@@ -50,6 +50,33 @@ pred=reloaded_model.predict(test_padded)
|
|
50 |
pred_bin = np.argmax(pred,1)
|
51 |
accuracy_score(pred_bin, sst2['test']['label'])
|
52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
|
54 |
```
|
55 |
|
|
|
50 |
pred_bin = np.argmax(pred,1)
|
51 |
accuracy_score(pred_bin, sst2['test']['label'])
|
52 |
|
53 |
+
0.9187259747391543
|
54 |
+
|
55 |
+
reloaded_model.summary()
|
56 |
+
|
57 |
+
Model: "model"
|
58 |
+
_________________________________________________________________
|
59 |
+
Layer (type) Output Shape Param #
|
60 |
+
=================================================================
|
61 |
+
input_1 (InputLayer) [(None, 64)] 0
|
62 |
+
|
63 |
+
embedding (Embedding) (None, 64, 50) 500000
|
64 |
+
|
65 |
+
bidirectional (Bidirectiona (None, 64, 128) 58880
|
66 |
+
l)
|
67 |
+
|
68 |
+
bidirectional_1 (Bidirectio (None, 128) 98816
|
69 |
+
nal)
|
70 |
+
|
71 |
+
dropout (Dropout) (None, 128) 0
|
72 |
+
|
73 |
+
dense (Dense) (None, 2) 258
|
74 |
+
|
75 |
+
=================================================================
|
76 |
+
Total params: 657,954
|
77 |
+
Trainable params: 657,954
|
78 |
+
Non-trainable params: 0
|
79 |
+
_________________________________________________________________
|
80 |
|
81 |
```
|
82 |
|