fix: 补充 BaseEstimator/TransformerMixin 导入

This commit is contained in:
duxin
2026-07-29 10:20:57 +08:00
parent 12866f12b0
commit 80189a9f44

View File

@ -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