add interface

8b6e6b02 · 赵威 · 865d27eb · 8b6e6b02
Commit 8b6e6b02 authored Oct 14, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

word_to_vec.py word_vector/word_to_vec.py +2 -2

No files found.
--- a/word_vector/word_to_vec.py
+++ b/word_vector/word_to_vec.py
@@ -35,8 +35,8 @@ def w2v_train(f_name, model_output_name):
 @bind("strategy_embedding/word_vector/word_similarity")
-def word_similarity():
+def word_similarity(word):
-    pass
+    return WORD2VEC_MODEL.wv.most_similar(word)
 if __name__ == "__main__":