소스 검색

feat:添加测试脚本

zhaohaipeng 5 달 전
부모
커밋
35f1534ab9
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      ad/model_predict_analyse.py

+ 2 - 2
ad/model_predict_analyse.py

@@ -132,8 +132,8 @@ def read_and_calibration_predict(predict_path: str, is_hdfs=True, step=100) -> [
 
 
 def _main(old_predict_path: str, new_predict_path: str, calibration_file: str, analyse_file: str):
-    old_group_df, old_segment_df = read_and_calibration_predict(old_predict_path, is_hdfs=False)
-    new_group_df, new_segment_df = read_and_calibration_predict(new_predict_path, is_hdfs=False)
+    old_group_df, old_segment_df = read_and_calibration_predict(old_predict_path, is_hdfs=True)
+    new_group_df, new_segment_df = read_and_calibration_predict(new_predict_path, is_hdfs=True)
 
     # 分段文件保存, 此处保留的最后使用的分段文件,不是所有的分段
     new_segment_df.to_csv(calibration_file, sep='\t', index=False, header=False)