Support loading gensim word2vec with new format.

wuyunxiangwyx · Sep 26, 2017 · 03132ff · 03132ff
1 parent ee15cd9
commit 03132ff
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Chatbot/RuleMatcher/rulebase.py b/Chatbot/RuleMatcher/rulebase.py
@@ -209,8 +209,10 @@ def load_model(self,path):
         Args:
             path: the path of the model.
         """
-
-        self.model = models.KeyedVectors.load_word2vec_format(path,binary=True)
+        try:
+            self.model = models.Word2Vec.load(path)  # cur
+        except:
+            self.model = models.KeyedVectors.load_word2vec_format(path, binary=True)  # old
 
     def match(self, sentence, topk=1, threshold=0, root=None):