[Fixed] issue#30

zhanggaoxuan · Jan 12, 2018 · 45f4e31 · 45f4e31
1 parent 03132ff
commit 45f4e31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Chatbot/model/test.py b/Chatbot/model/test.py
@@ -6,7 +6,7 @@
 
 def main():
 	logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s', level=logging.INFO)
-	model = models.Word2Vec.load_word2vec_format('ch-corpus-3sg.bin',binary=True)
+	model = models.Word2Vec.load('your_w2v_model_path')
 
 	print("提供 3 種測試模式\n")
 	print("輸入一個詞，則去尋找前一百個該詞的相似詞")
@@ -38,4 +38,4 @@ def main():
 			print(repr(e))
 
 if __name__ == "__main__":
-	main()
+	main()