add timer

word_vector/api.py

@@ -115,9 +115,13 @@ def clicked_tractate_ids_item2vec_model_by_ids(ids, n=5):
     try:
         time_begin = time.time()
         res = []
+        search_time_lst = []
         for id in ids:
             if id in TRACTATE_CLICK_IDS:
+                search_time_begin = time.time()
                 res.extend(TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n))
+                search_time_end = time.time() - search_time_begin
+                search_time_lst.append("{:.2f}".format(search_time_end * 1000))
         time_end = time.time() - time_begin
         if time_end > 0.04:
             timer_dict = {
@@ -125,7 +129,8 @@ def clicked_tractate_ids_item2vec_model_by_ids(ids, n=5):
                 "api": "strategy_embedding/word_vector/tractate_item2vec_by_ids",
                 "ids": ids,
                 "n": n,
-                "total_timer": "{:.3f}ms".format(time_end * 1000)
+                "search_time": search_time_lst,
+                "total_time": "{:.3f}ms".format(time_end * 1000)
             }
             msg_res = ""
             for (k, v) in timer_dict.items():