correct aggregation_method

scikit-learn-contrib · Dec 12, 2024 · 1693dd2 · 1693dd2
1 parent 506c25a
commit 1693dd2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mapie_v1/regression.py b/mapie_v1/regression.py
@@ -602,7 +602,10 @@ class JackknifeAfterBootstrapRegressor:
     Examples
     --------
     >>> regressor = JackknifeAfterBootstrapRegressor(
-    ...    estimator=LinearRegression(), confidence_level=0.9, n_bootstraps=8)
+    ...    estimator=LinearRegression(), 
+    ...    confidence_level=0.9, 
+    ...    resampling=8,
+    ...    aggregation_method="mean")
     >>> regressor.fit(X_train, y_train)
     >>> regressor.conformalize(X_conf, y_conf)
     >>> intervals = regressor.predict_set(X_test)