Update word2vec_features.py

referencing SLIM version of word2vec
johnjoo1 · Dec 12, 2018 · a8fa29f · a8fa29f
1 parent d3e1bcc
commit a8fa29f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/features/word2vec_features.py b/src/features/word2vec_features.py
@@ -10,7 +10,7 @@
 
 
 from gensim import models
-model2 = models.KeyedVectors.load_word2vec_format('data/external/GoogleNews-vectors-negative300.bin', binary=True)
+model2 = models.KeyedVectors.load_word2vec_format('data/external/GoogleNews-vectors-negative300-SLIM.bin', binary=True)
 
 
 from nltk.tokenize import RegexpTokenizer
@@ -71,4 +71,4 @@
 with open('models/mlb.pkl','wb') as f:
     pickle.dump(mlb,f)
 
-os.remove("data/external/GoogleNews-vectors-negative300.bin")
+os.remove("data/external/GoogleNews-vectors-negative300-SLIM.bin")