测试修改

This commit is contained in:
DXC
2026-06-23 14:39:54 +08:00
parent c4aa246c95
commit b8d263e494
21 changed files with 595 additions and 2706 deletions

View File

@ -32,13 +32,22 @@ class Step9MlPredictHandler(BaseStepHandler):
models_dir = config.get('models_dir') or str(context.models_dir)
# 【硬编码路径清除】优先读取前端 config 中的 output_path / output_dir
# 绝不允许私自拼接 11_12_13_predictions 覆盖用户在前端填写的路径。
# 前端 step9_ml_predict_panel.get_config() 用的是 'output_path' 这个 key。
output_dir = (
config.get('output_path')
or config.get('output_dir')
or str(context.prediction_dir / "9_ML_Prediction")
)
try:
result = PredictionStep.predict_water_quality(
sampling_csv_path=sampling_csv_path,
models_dir=models_dir,
metric=config.get('metric', 'test_r2'),
prediction_column=config.get('prediction_column', 'prediction'),
output_dir=str(context.prediction_dir / "9_ML_Prediction"),
output_dir=output_dir,
_report_generator=context.report_generator,
_external_model=config.get('_external_model'),
_external_model_path=config.get('_external_model_path'),