From 4adfbfb70c8814f36d476c179c419616b96ce80c Mon Sep 17 00:00:00 2001 From: shinnar Date: Thu, 7 Jan 2021 21:16:35 -0500 Subject: [PATCH] Fix text in description of refit parameter Of HalvingGridSearchCV and HalvingRandomSearchCV to refer to the respective classes --- sklearn/model_selection/_search_successive_halving.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/model_selection/_search_successive_halving.py b/sklearn/model_selection/_search_successive_halving.py index 512595b1943ce..b522ce7fbda41 100644 --- a/sklearn/model_selection/_search_successive_halving.py +++ b/sklearn/model_selection/_search_successive_halving.py @@ -448,7 +448,7 @@ class HalvingGridSearchCV(BaseSuccessiveHalving): The refitted estimator is made available at the ``best_estimator_`` attribute and permits using ``predict`` directly on this - ``GridSearchCV`` instance. + ``HalvingGridSearchCV`` instance. error_score : 'raise' or numeric Value to assign to the score if an error occurs in estimator fitting. @@ -735,7 +735,7 @@ class HalvingRandomSearchCV(BaseSuccessiveHalving): The refitted estimator is made available at the ``best_estimator_`` attribute and permits using ``predict`` directly on this - ``GridSearchCV`` instance. + ``HalvingRandomSearchCV`` instance. error_score : 'raise' or numeric Value to assign to the score if an error occurs in estimator fitting.