Fix 'Deprecated:gensim.models.Word2Vec.load_word2vec_format' with 'ge…

…nsim.models.KeyedVectors.load_word2vec_format' instead.
Cisco-Talos · Jun 26, 2018 · 08e5a12 · 08e5a12
1 parent 304cf47
commit 08e5a12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tree_model/Word2VecFeatureGenerator.py b/tree_model/Word2VecFeatureGenerator.py
@@ -27,7 +27,7 @@ def process(self, df):
 
         # 1). document vector built by multiplying together all the word vectors
         # using Google's pre-trained word vectors
-        model = gensim.models.Word2Vec.load_word2vec_format('GoogleNews-vectors-negative300.bin', binary=True)
+        model = gensim.models.KeyedVectors.load_word2vec_format('GoogleNews-vectors-negative300.bin', binary=True)
         print 'model loaded'
 
         Headline_unigram_array = df['Headline_unigram_vec'].values