Commit 8ab9ac24 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added hyperband option to hyperparameter tuning pipeline

parent 868b7056
Pipeline #88008 passed with stage
in 15 minutes and 22 seconds
......@@ -429,7 +429,7 @@ else:
"val_mae", patience=15, restore_best_weights=True
),
],
n_replicas=3,
n_replicas=1,
n_epochs=30,
)
......
......@@ -9,7 +9,7 @@ Simple utility functions used in deepof example scripts. These are not part of t
"""
from datetime import date, datetime
from kerastuner import BayesianOptimization
from kerastuner import BayesianOptimization, Hyperband
from kerastuner import HyperParameters
from kerastuner_tensorboard_logger import TensorBoardLogger
from typing import Tuple, Union, Any, List
......@@ -172,12 +172,12 @@ def tune_search(
else:
return False
tuner = BayesianOptimization(
tuner = Hyperband(
hypermodel,
directory="BayesianOptx_{}_{}".format(loss, str(date.today())),
directory="HyperBandx_{}_{}".format(loss, str(date.today())),
executions_per_trial=n_replicas,
logger=TensorBoardLogger(metrics=["val_mae"], logdir="./logs/hparams"),
max_trials=bayopt_trials,
max_epochs=bayopt_trials,
objective="val_mae",
project_name=project_name,
seed=42,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment