From 1693dd2b03432295cb9db27c2643aafdb4af3a86 Mon Sep 17 00:00:00 2001 From: qroa Date: Thu, 12 Dec 2024 16:49:05 +0100 Subject: [PATCH] correct aggregation_method --- mapie_v1/regression.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mapie_v1/regression.py b/mapie_v1/regression.py index 72d1c9183..e6ad73a16 100644 --- 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)