Commit 5123d2da authored by 赵威's avatar 赵威

update timer

parent a5357712
......@@ -92,8 +92,20 @@ def clicked_tractate_ids_item2vec_model(id, n=5):
res = 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 id={} n={} cost {:.3f}ms".format(
id, n, time_end * 1000))
timer_dict = {
"method": "clicked_tractate_ids_item2vec_model",
"api": "strategy_embedding/word_vector/tractate_item2vec",
"id": id,
"n": n,
"total_time": "{:.3f}ms".format(time_end * 1000)
}
msg_res = ""
for (k, v) in timer_dict.items():
msg_res += str(k)
msg_res += ": "
msg_res += str(v)
msg_res += "\n"
send_performance_msg_to_dingtalk(msg_res)
return res
except KeyError as e:
# send_msg_to_dingtalk("tractate_item2vec: " + str(e))
......
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