Browse Source

add ad_out

baichongyang 1 year ago
parent
commit
465d6f522a
1 changed files with 27 additions and 15 deletions
  1. 27 15
      ad_out_v1_set_config.py

+ 27 - 15
ad_out_v1_set_config.py

@@ -11,20 +11,32 @@ redis_helper = RedisHelper()
 
 if __name__ == '__main__':
     model_key = 'ad_out_v1'
-    abtest_id = '173'
-    abtest_config_tag_list = ['u', 'v']
-    for abtest_config_tag in abtest_config_tag_list :
+    configs = {
+        '173-u': {
+            'threshold': 0.26,
+            'item_threshold': 0.3957,
+            'user_threshold': 0.3272,
+            'miss_threshold': 0.0,
+            'use_backup': 'true'
+        },
+        '173-v': {
+            'threshold': 0.2705,
+            'item_threshold': 0.3957,
+            'user_threshold': 0.3272,
+            'miss_threshold': 0.0,
+            'use_backup': 'false'
+        },
+    }
+    # abtest_id = '173'
+    # abtest_config_tag_list = ['u', 'v']
+    expire_time = 30 * 24 * 3600
+    for abtest_key, abtest_config in configs.items():
+        print(abtest_key)
+        abtest_id, abtest_config_tag = abtest_key.split('-')
         config_key_prefix = f"{config_.KEY_NAME_PREFIX_AD_OUT_MODEL_CONFIG}{model_key}:{abtest_id}:{abtest_config_tag}"
-        threshold_key = f"{config_key_prefix}:threshold"
-        use_mean_key = f"{config_key_prefix}:use_mean"
-        print(threshold_key)
-        threshold = redis_helper.get_data_from_redis(key_name=threshold_key)
-        use_mean = redis_helper.get_data_from_redis(key_name=use_mean_key)
-        print('threshold:', threshold, type(threshold))
-        print('use_mean:', use_mean, type(use_mean))
-        expire_time = 30 * 24 * 3600
-        redis_helper.set_data_to_redis(threshold_key, 0.27, expire_time)
-        redis_helper.set_data_to_redis(use_mean_key, 'false', expire_time)
-        print(threshold_key)
-        print(use_mean_key)
+        for k, v in abtest_config.items():
+            k_key = f"{config_key_prefix}:{k}"
+            k_value = redis_helper.get_data_from_redis(key_name=k_key)
+            print(k_key, k_value, type(k_value))
+            redis_helper.set_data_to_redis(k_key, v, expire_time)