Get weighted sum of trained layers

CyberZHG · Oct 30, 2018 · de2a67a · de2a67a
1 parent ec29261
commit de2a67a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -93,7 +93,8 @@ bi_lm = BiLM(token_num=20000,
              rnn_layer_num=3,
              rnn_keep_num=4,
              rnn_units=300,
-             rnn_type='lstm')
+             rnn_type='lstm',
+             use_normalization=True)
 # ...
 # Train the Bi-LM model
 # ...

diff --git a/tests/test_get_feature_layers.py b/tests/test_get_feature_layers.py
@@ -73,7 +73,8 @@ def test_weighted_sum(self):
                      rnn_layer_num=6,
                      rnn_keep_num=7,
                      rnn_units=108,
-                     rnn_type='lstm')
+                     rnn_type='lstm',
+                     use_normalization=True)
         input_layer, output_layer = bi_lm.get_feature_layers(use_weighted_sum=True)
         model = keras.models.Model(inputs=input_layer, outputs=output_layer)
         model.summary()