diff --git a/src/core/modeling/modeling_batch.py b/src/core/modeling/modeling_batch.py index 8f7f79a..17378fd 100644 --- a/src/core/modeling/modeling_batch.py +++ b/src/core/modeling/modeling_batch.py @@ -10,6 +10,7 @@ warnings.filterwarnings('ignore') # 机器学习模型导入 - 改为回归模型 from sklearn.base import BaseEstimator, TransformerMixin +from sklearn.preprocessing import StandardScaler from sklearn.svm import SVR from sklearn.ensemble import RandomForestRegressor from sklearn.neighbors import KNeighborsRegressor