Explorar o código

feat:添加模型分本地保存逻辑

zhaohaipeng hai 5 meses
pai
achega
e5592c4144
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      ad/model_predict_analyse.py

+ 2 - 2
ad/model_predict_analyse.py

@@ -135,9 +135,9 @@ def predict_local_save_for_auc(old_df: pd.DataFrame, new_df: pd.DataFrame):
     d = {"old": old_df, "new": new_df}
     for key in d:
         df = d[key][['label', "score"]]
-        df.to_csv(f"{PREDICT_CACHE_PATH}/{key}_1.csv", index=False, header=False)
+        df.to_csv(f"{PREDICT_CACHE_PATH}/{key}_1.csv", sep="\t", index=False, header=False)
         df = d[key][['label', "score_2"]]
-        df.to_csv(f"{PREDICT_CACHE_PATH}/{key}_2.csv", index=False, header=False)
+        df.to_csv(f"{PREDICT_CACHE_PATH}/{key}_2.csv", sep="\t", index=False, header=False)
 
 
 def _main(old_predict_path: str, new_predict_path: str, calibration_file: str, analyse_file: str):