测试修改
This commit is contained in:
@ -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'),
|
||||
|
||||
Reference in New Issue
Block a user