From 2975d91ca4b375b1e88888b9bb420b25e1c3d1d8 Mon Sep 17 00:00:00 2001 From: Ruan Comelli Date: Wed, 6 Jul 2022 14:43:15 -0300 Subject: [PATCH] fix(automl/tuners): fix inheritance chain --- boiling_learning/automl/tuners.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/boiling_learning/automl/tuners.py b/boiling_learning/automl/tuners.py index c6eeab8a..7240179e 100644 --- a/boiling_learning/automl/tuners.py +++ b/boiling_learning/automl/tuners.py @@ -41,7 +41,7 @@ def set_state(self, state: Dict[str, Any]) -> None: self.stop_search = state['stop_search'] -class EarlyStoppingHyperbandOracle(ak.tuners.hyperband.HyperbandOracle): +class EarlyStoppingHyperbandOracle(kt.oracles.HyperbandOracle): def __init__(self, *args: Any, **kwargs: Any) -> None: super().__init__(*args, **kwargs) self.stop_search = False @@ -227,6 +227,7 @@ def __init__( goal: Any, objective: str = 'val_loss', max_epochs: int = 100, + max_trials: int = 1000, factor: int = 3, seed: Optional[int] = None, hyperparameters: Optional[kt.HyperParameters] = None, @@ -244,6 +245,7 @@ def __init__( tune_new_entries=tune_new_entries, allow_new_entries=allow_new_entries, ) + oracle.max_trials = max_trials super().__init__(oracle=oracle, **kwargs) def on_epoch_end(