27 lines
1.3 KiB
Diff
27 lines
1.3 KiB
Diff
From b274e3e2455df84a95ed34acd118dd72c5e14313 Mon Sep 17 00:00:00 2001
|
|
From: dongjiao <dongjiao@kylinos.cn>
|
|
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
|
|
|