Forráskód Böngészése

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

zhaohaipeng 7 hónapja
szülő
commit
42422a579c
1 módosított fájl, 1 hozzáadás és 1 törlés
  1. 1 1
      model/model_predict_analyse_20241115.py

+ 1 - 1
model/model_predict_analyse_20241115.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)