From b274e3e2455df84a95ed34acd118dd72c5e14313 Mon Sep 17 00:00:00 2001 From: dongjiao Date: Tue, 11 Mar 2025 11:09:07 +0800 Subject: [PATCH] scikit-learn-1.2 rename the parameter 'base_estimator' to 'estimator'. --- analysis/optimizer/weighted_ensemble_feature_selector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/analysis/optimizer/weighted_ensemble_feature_selector.py b/analysis/optimizer/weighted_ensemble_feature_selector.py index cb39561..4137fe4 100644 --- a/analysis/optimizer/weighted_ensemble_feature_selector.py +++ b/analysis/optimizer/weighted_ensemble_feature_selector.py @@ -86,7 +86,7 @@ class WeightedEnsembleFeatureSelector: graboo = GradientBoostingRegressor(n_estimators=10000, learning_rate=0.1) adb = AdaBoostRegressor(DecisionTreeRegressor(max_depth=16), n_estimators=10000, random_state=0) - bag = BaggingRegressor(base_estimator=ExtraTreeRegressor(max_depth=16), + bag = BaggingRegressor(estimator=ExtraTreeRegressor(max_depth=16), n_estimators=10000, random_state=0, n_jobs=-1) self._regressors = [dtree, ranfor, graboo, adb, bag] self._ensemble_model = Ridge(alpha=10, max_iter=1000000) -- 2.43.0