Browse Source

add base rank

linfan 1 year ago
parent
commit
f7a00aeb25
2 changed files with 32 additions and 4 deletions
  1. 10 4
      config.py
  2. 22 0
      recommend.py

+ 10 - 4
config.py

@@ -145,7 +145,9 @@ class BaseConfig(object):
             'abtest_262': 60046,
             'abtest_316': 60047,
             'abtest_319': 60048,
-            'abtest_320': 60049
+            'abtest_320': 60049,
+            'abtest_322': 60050,
+            'abtest_323': 60051,
         },  # 地域分组小时级规则实验
 
         'rank_by_24h': {
@@ -330,9 +332,13 @@ class BaseConfig(object):
             'data_key': 'data10', 'rule_key': 'rule19',
             'ab_code': AB_CODE['region_rank_by_h'].get('abtest_319')
         },
-        '320': {
-            'data_key': 'data10', 'rule_key': 'rule19',
-            'ab_code': AB_CODE['region_rank_by_h'].get('abtest_320')
+        '322': {
+            'data_key': 'data10', 'rule_key': 'rule7',
+            'ab_code': AB_CODE['region_rank_by_h'].get('abtest_322')
+        },
+        '323': {
+            'data_key': 'data10', 'rule_key': 'rule7',
+            'ab_code': AB_CODE['region_rank_by_h'].get('abtest_323                                                                                                                                              ')
         }
     }
 

+ 22 - 0
recommend.py

@@ -1342,6 +1342,15 @@ def video_homepage_recommend(request_id, mid, uid, size, app_type, algo_type,
                              rule_key=rule_key, data_key=data_key,
                              no_op_flag=no_op_flag, old_video_index=old_video_index,
                              params=params, rule_key_30day=rule_key_30day, shield_config=shield_config)
+    elif ab_code == 60050 or ab_code == 60051:
+        result = video_recommend(request_id=request_id,
+                                 mid=mid, uid=uid, app_type=app_type,
+                                 size=size, top_K=top_K, flow_pool_P=flow_pool_P,
+                                 algo_type=algo_type, client_info=client_info,
+                                 ab_code=ab_code, expire_time=expire_time,
+                                 rule_key=rule_key, data_key=data_key,
+                                 no_op_flag=no_op_flag, old_video_index=old_video_index,
+                                 params=params, rule_key_30day=rule_key_30day, shield_config=shield_config)
     else:
         result = video_recommend(request_id=request_id,
                              mid=mid, uid=uid, app_type=app_type,
@@ -1373,6 +1382,8 @@ def video_homepage_recommend(request_id, mid, uid, size, app_type, algo_type,
     update_redis_st = time.time()
     if ab_code == 60047 or  ab_code == 60048 or  ab_code == 60049:
         update_flow_redis_data(result=rank_result, app_type=app_type, mid=mid, top_K=top_K)
+    elif ab_code == 60050 or  ab_code == 60051:
+        update_redis_data(result=rank_result, app_type=app_type, mid=mid, top_K=top_K)
     else:
         update_redis_data(result=rank_result, app_type=app_type, mid=mid, top_K=top_K)
     # log_.info({
@@ -1446,6 +1457,15 @@ def video_relevant_recommend(request_id, video_id, mid, uid, size, app_type, ab_
                                  old_video_index=old_video_index, video_id=video_id,
                                  params=params, rule_key_30day=rule_key_30day, shield_config=shield_config)
         # log_.info({
+    elif ab_code == 60050 or ab_code == 60051:
+        result = video_recommend(request_id=request_id,
+                                 mid=mid, uid=uid, app_type=app_type,
+                                 size=size, top_K=top_K, flow_pool_P=flow_pool_P,
+                                 algo_type='', client_info=client_info,
+                                 ab_code=ab_code, expire_time=expire_time,
+                                 rule_key=rule_key, data_key=data_key, no_op_flag=no_op_flag,
+                                 old_video_index=old_video_index, video_id=video_id,
+                                 params=params, rule_key_30day=rule_key_30day, shield_config=shield_config)
     else:
         result = video_recommend(request_id=request_id,
                              mid=mid, uid=uid, app_type=app_type,
@@ -1478,6 +1498,8 @@ def video_relevant_recommend(request_id, video_id, mid, uid, size, app_type, ab_
     update_redis_st = time.time()
     if ab_code == 60047 or ab_code == 60048 or  ab_code == 60049:
         update_flow_redis_data(result=rank_result, app_type=app_type, mid=mid, top_K=top_K)
+    elif ab_code == 60050 or ab_code == 60051:
+        update_redis_data(result=rank_result, app_type=app_type, mid=mid, top_K=top_K)
     else:
         update_redis_data(result=rank_result, app_type=app_type, mid=mid, top_K=top_K)
     # log_.info({