Update api_nn.py

using SLIM word2vec
johnjoo1 · Dec 12, 2018 · cc267dd · cc267dd
1 parent a8fa29f
commit cc267dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/apis/api_nn.py b/src/apis/api_nn.py
@@ -13,7 +13,7 @@
     Genre_ID_to_name=pickle.load(f)
 
 model_textual = load_model('models/overview_nn.h5')
-w2v_model = models.KeyedVectors.load_word2vec_format('data/external/GoogleNews-vectors-negative300.bin', binary=True)
+w2v_model = models.KeyedVectors.load_word2vec_format('data/external/GoogleNews-vectors-negative300-SLIM.bin', binary=True)
 tokenizer = RegexpTokenizer(r'\w+')
 en_stop = get_stop_words('en')
 with open('models/mlb.pkl','rb') as f: