added word embeddings file

This commit is contained in:
codekansas
2016-05-04 11:40:21 -04:00
parent 2226ef0b90
commit 894dc21914
+1 -1
View File
@@ -264,7 +264,7 @@ if __name__ == '__main__':
# np.save(open('models/embedding_200_dim.h5', 'wb'), weights)
# load pre-trained embedding layer
weights = np.load('models/word2vec_100_dim.h5')
weights = np.load('word2vec_100_dim.h5')
language_model = model.prediction_model.layers[2]
language_model.layers[2].set_weights([weights])