item2vec for answer

a68c0f19 · 赵威 · 0b4b9c6c · a68c0f19
Commit a68c0f19 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
@@ -189,7 +189,7 @@ if __name__ == "__main__":
    save_clicked_answer_ids_item2vec()

    for id in ["986424", "744910", "703622"]:
-        print(ANSWER_CLICK_IDS_MODEL.wv.most_similar(id, topn=5))
+        print(ANSWER_CLICK_IDS_MODEL.wv.most_similar(id, topn=10))

    print("total cost: {:.2f}mins".format((time.time() - begin_time) / 60))