浏览代码

feat:添加测试脚本

zhaohaipeng 5 月之前
父节点
当前提交
87ac5a0316
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      ad/model_predict_analyse.py

+ 1 - 1
ad/model_predict_analyse.py

@@ -85,7 +85,7 @@ def segment_calc_diff_rate_by_score(df: pd.DataFrame, segment_file_path: str, st
 
     # 完整的分段文件保存
     csv_data = group_df.to_csv(sep="\t", index=False)
-    with client.write(segment_file_path, encoding='utf08') as writer:
+    with client.write(segment_file_path, encoding='utf-8') as writer:
         writer.write(csv_data)
 
     filtered_df = group_df[(abs(group_df['segment_diff_rate']) >= 0.2) & (group_df['segment_label_cnt'] >= 1000)]