|
|
@@ -133,11 +133,12 @@ def predict():
|
|
|
for config_key, config_param in params.items():
|
|
|
predict_ad_group_video(dt=dt, config_key=config_key, config_param=config_param)
|
|
|
# 阈值参数记录
|
|
|
- # for abtest_group, ad_threshold_mapping in config_param['threshold'].items():
|
|
|
- # threshold_record[f"{config_key}-{abtest_group}"] = ad_threshold_mapping['group']
|
|
|
- # redis_helper.set_data_to_redis(key_name=config_.KEY_NAME_PREFIX_AD_THRESHOLD_RECORD,
|
|
|
- # value=str(threshold_record),
|
|
|
- # expire_time=24*3600)
|
|
|
+ ad_threshold_mappings = config_.AD_ABTEST_THRESHOLD_CONFIG.get(config_key.split('-')[0])
|
|
|
+ for abtest_group, ad_threshold_mapping in ad_threshold_mappings.items():
|
|
|
+ threshold_record[f"{config_key}-{abtest_group}"] = ad_threshold_mapping['group']
|
|
|
+ redis_helper.set_data_to_redis(key_name=config_.KEY_NAME_PREFIX_AD_THRESHOLD_RECORD,
|
|
|
+ value=str(threshold_record),
|
|
|
+ expire_time=24*3600)
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
