Set trainable flag in get_embedding_layer. Fix 1557 (#1558)

piskvorky · Sep 1, 2017 · 3d2227d · 3d2227d
1 parent 9caf055
commit 3d2227d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -832,5 +832,8 @@ def get_embedding_layer(self, train_embeddings=False):
         if not KERAS_INSTALLED:
             raise ImportError("Please install Keras to use this function")
         weights = self.syn0
-        layer = Embedding(input_dim=weights.shape[0], output_dim=weights.shape[1], weights=[weights])  # No extra mem usage here as `Embedding` layer doesn't create any new matrix for weights
+
+        # set `trainable` as `False` to use the pretrained word embedding
+        # No extra mem usage here as `Embedding` layer doesn't create any new matrix for weights
+        layer = Embedding(input_dim=weights.shape[0], output_dim=weights.shape[1], weights=[weights], trainable=train_embeddings)
         return layer