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):
res = []
for id in 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
if time_end > 0.04:
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