add some ids

bd12e9a0 · 赵威 · e1f21fe3 · bd12e9a0
Commit bd12e9a0 authored Dec 22, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

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

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

    save_clicked_answer_ids_item2vec()

-    for id in ["744910", "507016", "799682", "712841"]:
+    for id in ["744910", "507016", "799682", "712841", "990793", "753792", "862143"]:
        try:
+            ANSWER_CLICK_IDS_MODEL = word2vec.Word2Vec.load(answer_click_ids_model_path)
+            ANSWER_CLICK_IDS = set(ANSWER_CLICK_IDS_MODEL.wv.vocab.keys())
            print(ANSWER_CLICK_IDS_MODEL.wv.most_similar(id, topn=10))
        except Exception as e:
            print(e)