Skip to content

S
strategy_embedding
Commit 79674b06 authored by 赵威's avatar 赵威
Browse files
Options

add timer

parent 5107215f
Hide whitespace changes
Inline Side-by-side
Showing with 16 additions and 16 deletions
personas_vector/match_api.py
View file @ 79674b06
...@@ -42,22 +42,22 @@ def match_tractate_by_device(device_id, n=10): ...@@ -42,22 +42,22 @@ def match_tractate_by_device(device_id, n=10):
if i <= 5.0: if i <= 5.0:
res.append(ids[index]) res.append(ids[index])
time_end = time.time() - time_begin time_end = time.time() - time_begin
# if time_end > 0.04: if time_end > 0.04:
timer_dict = { timer_dict = {
"method": "match_tractate_by_device", "method": "match_tractate_by_device",
"api": "strategy_embedding/personas_vector/match", "api": "strategy_embedding/personas_vector/match",
"device_id": device_id, "device_id": device_id,
"n": n, "n": n,
"search": "{:.3f}ms".format(search_time_end * 1000), "search": "{:.3f}ms".format(search_time_end * 1000),
"total_time": "{:.3f}ms".format(time_end * 1000) "total_time": "{:.3f}ms".format(time_end * 1000)
} }
msg_res = "" msg_res = ""
for (k, v) in timer_dict.items(): for (k, v) in timer_dict.items():
msg_res += str(k) msg_res += str(k)
msg_res += ": " msg_res += ": "
msg_res += str(v) msg_res += str(v)
msg_res += "\n" msg_res += "\n"
send_performance_msg_to_dingtalk(msg_res) send_performance_msg_to_dingtalk(msg_res)
return res return res
return [] return []
except Exception as e: except Exception as 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