fix: 补充 BaseEstimator/TransformerMixin 导入
This commit is contained in:
@ -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
|
||||
|
||||
Reference in New Issue
Block a user