liqian hace 2 años
padre
commit
0497d12bb9
Se han modificado 2 ficheros con 10 adiciones y 1 borrados
  1. 8 1
      ad_recommend.py
  2. 2 0
      config.py

+ 8 - 1
ad_recommend.py

@@ -36,8 +36,15 @@ def ad_recommend_predict(mid, video_id):
 
     # 计算 mid-video 分享率
     mid_video_share_rate = float(group_share_rate) * float(video_share_rate)
+
+    # 获取对应的阈值
+    threshold_key_name = f"{config_.KEY_NAME_PREFIX_AD_THRESHOLD}{mid_group}"
+    threshold = redis_helper.get_data_from_redis(key_name=threshold_key_name)
+    if threshold is None:
+        threshold = 0
+    else:
+        threshold = float(threshold)
     # 阈值判断
-    threshold = 0.05
     if mid_video_share_rate > threshold:
         ad_predict = 1
     else:

+ 2 - 0
config.py

@@ -616,6 +616,8 @@ class BaseConfig(object):
     KEY_NAME_PREFIX_AD_VIDEO = 'ad:video:predict:share:rate:'
     # 用户分组结果存放 redis key 前缀,完整格式:mid:group:{mid}
     KEY_NAME_PREFIX_MID_GROUP = 'mid:group:'
+    # 广告推荐阈值结果存放 redis key 前缀,完整格式:ad:threshold:{group}
+    KEY_NAME_PREFIX_AD_THRESHOLD = 'ad:threshold:'
 
 
 class DevelopmentConfig(BaseConfig):