diff --git a/insurance_qa_eval.py b/insurance_qa_eval.py index 3d9fe36..b73ceb4 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('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])