update

ad_thompson_param_update.py

@@ -14,6 +14,8 @@ log_ = Log()
 
 def main(project, table):
     try:
+        view_pv_sum = 0
+        click_pv_sum = 0
         # 获取广告ad_idea_id对应的曝光次数和点击次数
         odps = ODPS(
             access_id=config_.ODPS_CONFIG['ACCESSID'],
@@ -42,6 +44,8 @@ def main(project, table):
             param_alpha = click_pv
             param_beta = view_pv - click_pv
             ad_idea_data.append({'ad_idea_id': ad_idea_id, 'param': [param_alpha, param_beta]})
+            view_pv_sum += view_pv
+            click_pv_sum += click_pv
         log_.info(f"ad_idea_data count: {len(ad_idea_data)}")
         log_.info(f"ad_idea_data: {ad_idea_data}")
         # 更新redis
@@ -55,6 +59,11 @@ def main(project, table):
                 i += 1
             except:
                 continue
+        # 更新均值
+        key_name = f"{config_.THOMPSON_PARAM_KEY_PREFIX}-1"
+        value = json.dumps([int(click_pv_sum * 1.2), view_pv_sum])
+        redis_helper.set_data_to_redis(key_name=key_name, value=value, expire_time=24 * 3600)
+        i += 1
         log_.info(f"to redis count: {i}")
     except Exception as e:
         log_.error(f"广告Thompson参数更新失败, exception: {e}, traceback: {traceback.format_exc()}")