fix deprecated base_estimator

scikit-learn-contrib · Jul 9, 2024 · 45b656d · 45b656d
1 parent 7903f00
commit 45b656d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/plot_mpg_svr.py b/examples/plot_mpg_svr.py
@@ -39,7 +39,7 @@
 # Create RandomForestRegressor
 n_estimators = 1000
 mpg_bagger = BaggingRegressor(
-    base_estimator=SVR(), n_estimators=n_estimators, random_state=42
+    estimator=SVR(), n_estimators=n_estimators, random_state=42
 )
 mpg_bagger.fit(mpg_X_train, mpg_y_train)
 mpg_y_hat = mpg_bagger.predict(mpg_X_test)

diff --git a/forestci/tests/test_forestci.py b/forestci/tests/test_forestci.py
@@ -91,7 +91,7 @@ def test_bagging_svr_error():
     X_train = X[train_idx]
 
     n_trees = 4
-    bagger = BaggingRegressor(base_estimator=SVR(), n_estimators=n_trees)
+    bagger = BaggingRegressor(estimator=SVR(), n_estimators=n_trees)
     bagger.fit(X_train, y_train)
     inbag = fci.calc_inbag(X_train.shape[0], bagger)
     for ib in [inbag, None]: