liqian 1 ano atrás
pai
commit
ff2d8bbb34
1 arquivos alterados com 5 adições e 0 exclusões
  1. 5 0
      ad_predict.py

+ 5 - 0
ad_predict.py

@@ -49,11 +49,15 @@ def get_creative_id_with_thompson(mid, creative_id_list):
     """利用Thompson采样获取此次要展示的广告创意ID"""
     # 限制协程最大并发数:20
     gevent_pool = pool.Pool(100)
+    st_time5 = time.time()
     tasks = [gevent_pool.spawn(thompson_process, creative_id) for creative_id in creative_id_list]
     gevent.joinall(tasks)
     thompson_res_list = [t.get() for t in tasks]
+    get_random_time = int(time.time() - st_time5) * 1000
     # 按照score排序
+    st_time4 = time.time()
     thompson_res_rank = sorted(thompson_res_list, key=lambda x: x[1], reverse=True)
+    sort_time = int(time.time() - st_time4) * 1000
     rank_res = {
         'mid': mid,
         'creative_id': thompson_res_rank[0][0],
@@ -62,5 +66,6 @@ def get_creative_id_with_thompson(mid, creative_id_list):
         'random_flag': thompson_res_rank[0][3],
         'thompson_res_rank': thompson_res_rank
     }
+    log_.info(f"get_random_time: {get_random_time}, sort_time: {sort_time}")
     return rank_res