From 960f53b70b676c11a298b2067200d1d769431f3e Mon Sep 17 00:00:00 2001 From: JoaoPedroNC <95342464+JoaoPedroNC@users.noreply.github.com> Date: Wed, 28 Jun 2023 19:30:47 -0300 Subject: [PATCH] =?UTF-8?q?adaptee.py=20atualizado=20com=20tipifica=C3=A7?= =?UTF-8?q?=C3=A3o?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Leo, verifique se essa tipificação está boa o bastante --- skga/adaptee.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/skga/adaptee.py b/skga/adaptee.py index 4552512..1824d73 100644 --- a/skga/adaptee.py +++ b/skga/adaptee.py @@ -17,7 +17,7 @@ class Decoder: - def __init__(self, parameters, estimator, X, y, cv): + def __init__(self, parameters: dict, estimator: float, X:float, y: float, cv: dict): self._parameters = parameters self._estimator = estimator self._X = X @@ -157,7 +157,7 @@ def fit(self, X, y=None, *, groups=None, **fit_params): cv_orig = check_cv(self.cv, y, classifier=is_classifier(estimator)) n_splits = cv_orig.get_n_splits(X, y, groups) - def evaluate_candidates(candidate_params, cv=None, more_results=None): + def evaluate_candidates(candidate_params: float, cv=None, more_results=None): start = datetime.now() candidate_params = list(candidate_params) all_candidate_params = [] @@ -217,5 +217,5 @@ def evaluate_candidates(candidate_params, cv=None, more_results=None): return self - def _run_search(self, evaluate_candidates): + def _run_search(self, evaluate_candidates: bool): evaluate_candidates(ParameterGrid(self._parameters))