From bda371a698531afbe7bf9b42c7010ffd8cb51048 Mon Sep 17 00:00:00 2001 From: Candice Moyet Date: Wed, 12 Jul 2023 09:53:07 +0200 Subject: [PATCH] UPD: add parameter's description in docstring --- mapie/estimator/estimator.py | 4 +++- mapie/estimator/interface.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/mapie/estimator/estimator.py b/mapie/estimator/estimator.py index ed05b850d..f96a32219 100644 --- a/mapie/estimator/estimator.py +++ b/mapie/estimator/estimator.py @@ -470,7 +470,9 @@ def predict( By default ``False``. return_multi_pred: bool - + If ``True`` the method returns the predictions and the multiple + predictions (3 arrays). If ``False`` the method return the + simple predictions only. Returns ------- diff --git a/mapie/estimator/interface.py b/mapie/estimator/interface.py index dafbb382d..468f33185 100644 --- a/mapie/estimator/interface.py +++ b/mapie/estimator/interface.py @@ -73,7 +73,9 @@ def predict( By default ``False``. return_multi_pred: bool - + If ``True`` the method returns the predictions and the multiple + predictions (3 arrays). If ``False`` the method return the + simple predictions only. Returns -------