Ver código fonte

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

zhaohaipeng 7 meses atrás
pai
commit
e64fae8d2f
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      model/model_predict_analyse_20241115.py

+ 1 - 1
model/model_predict_analyse_20241115.py

@@ -139,7 +139,7 @@ def _main(old_predict_path: str, new_predict_path: str, calibration_file: str, a
     predict_local_save_for_auc(old_df, new_df)
 
     # 新模型校准文件保存本地,用于同步OSS
-    new_calibration_df.to_csv(calibration_file, sep="\t", index=False)
+    new_calibration_df[['cid', 'diff_rate']].to_csv(calibration_file, sep="\t", index=False, header=False)
 
     old_agg_df = predict_df_agg(old_df)
     new_agg_df = predict_df_agg(new_df)