add ids

4cdb7529 · 赵威 · 87542a4f · 4cdb7529
Commit 4cdb7529 authored Dec 22, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

answer.py word_vector/answer.py +1 -1

No files found.
--- a/word_vector/answer.py
+++ b/word_vector/answer.py
@@ -195,7 +195,7 @@ if __name__ == "__main__":

    save_clicked_answer_ids_item2vec()

-    for id in ["507016"]:
+    for id in ["507016", "970022", "973804", "249063"]:
        try:
            ANSWER_CLICK_IDS_MODEL = word2vec.Word2Vec.load(answer_click_ids_model_path)
            print(ANSWER_CLICK_IDS_MODEL.wv.most_similar(id, topn=10))