moshew commited on
Commit
49039c7
·
1 Parent(s): cd4126b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
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