diff --git a/insurance_qa_eval.py b/insurance_qa_eval.py index b73ceb4..c63834a 100644 --- a/insurance_qa_eval.py +++ b/insurance_qa_eval.py @@ -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('word2vec_100_dim.h5') + weights = np.load('word2vec_100_dim.embeddings') language_model = model.prediction_model.layers[2] language_model.layers[2].set_weights([weights]) diff --git a/word2vec_100_dim.embeddings b/word2vec_100_dim.embeddings new file mode 100755 index 0000000..c693177 Binary files /dev/null and b/word2vec_100_dim.embeddings differ