diff --git a/src/core/modeling/modeling_batch.py b/src/core/modeling/modeling_batch.py index 0448246..5e27447 100644 --- a/src/core/modeling/modeling_batch.py +++ b/src/core/modeling/modeling_batch.py @@ -641,8 +641,10 @@ class WaterQualityModelingBatch: ]) # ★ 入口清洗:inf → NaN,后续 SimpleImputer/cleaner 接力处理 - X_train = X_train.replace([np.inf, -np.inf], np.nan) - X_test = X_test.replace([np.inf, -np.inf], np.nan) + X_train = np.asarray(X_train, dtype=np.float64) + X_test = np.asarray(X_test, dtype=np.float64) + X_train[~np.isfinite(X_train)] = np.nan + X_test[~np.isfinite(X_test)] = np.nan # RandomizedSearchCV 需要以「步骤名__参数名」的格式索引参数网格; # 我们原有的 config['params'] 是模型层的(无 __),统一加 model__ 前缀。