Procházet zdrojové kódy

feat:修改模型评估结果分析脚本

zhaohaipeng před 4 měsíci
rodič
revize
a82f3f5b5b
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      ad/model_predict_analyse.py

+ 1 - 1
ad/model_predict_analyse.py

@@ -82,7 +82,7 @@ def get_predict_calibration_file(df: pd.DataFrame, predict_basename: str) -> [pd
     计算模型分的diff_rate
     """
     agg_df = predict_df_agg(df)
-    agg_df['diff_rate'] = (agg_df['score_avg'] / agg_df['true_ctcvr'] - 1).mask(agg_df['true_ctcvr'] == 0, 0)
+    agg_df['diff_rate'] = (agg_df['score_avg'] / agg_df['true_ctcvr'] - 1).mask(agg_df['true_ctcvr'] == 0, 0).round(6)
     condition = 'view > 1000 and diff_rate >= 0.2'
     save_full_calibration_file(agg_df, f"{SEGMENT_BASE_PATH}/{predict_basename}.txt")
     calibration = agg_df.query(condition)