Fixing minor bug in pretrained_word_embeddings example (keras-team#9438)

senon · Feb 21, 2018 · 4b9ba1b · 4b9ba1b
1 parent 531a8ea
commit 4b9ba1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pretrained_word_embeddings.py b/examples/pretrained_word_embeddings.py
@@ -101,7 +101,7 @@
 print('Preparing embedding matrix.')
 
 # prepare embedding matrix
-num_words = min(MAX_NUM_WORDS, len(word_index))
+num_words = min(MAX_NUM_WORDS, len(word_index) + 1)
 embedding_matrix = np.zeros((num_words, EMBEDDING_DIM))
 for word, i in word_index.items():
     if i >= MAX_NUM_WORDS: