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