Commit 5982f258 authored by 赵威's avatar 赵威

update sim function

parent 91c45347
...@@ -117,7 +117,7 @@ def clicked_tractate_ids_item2vec_model_by_ids(ids, n=5): ...@@ -117,7 +117,7 @@ def clicked_tractate_ids_item2vec_model_by_ids(ids, n=5):
res = [] res = []
for id in ids: for id in ids:
if id in TRACTATE_CLICK_IDS: if id in TRACTATE_CLICK_IDS:
res.extend(clicked_tractate_ids_item2vec_model(id, n)) res.extend(TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n))
time_end = time.time() - time_begin time_end = time.time() - time_begin
if time_end > 0.04: if time_end > 0.04:
send_performance_msg_to_dingtalk("clicked_tractate_ids_item2vec_model_by_ids ids={} n={} cost {:.3f}ms".format( send_performance_msg_to_dingtalk("clicked_tractate_ids_item2vec_model_by_ids ids={} n={} cost {:.3f}ms".format(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment