From 8fc1694a95b9ed21926e52e557c5e71813b406cf Mon Sep 17 00:00:00 2001 From: Makoto Uchida Date: Tue, 16 Jun 2020 13:16:26 -0700 Subject: [PATCH] Pass down tuner_id from Tuner class constractor (#269) --- kerastuner/engine/tuner.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/kerastuner/engine/tuner.py b/kerastuner/engine/tuner.py index da7a05b0c..9eee94cee 100644 --- a/kerastuner/engine/tuner.py +++ b/kerastuner/engine/tuner.py @@ -67,6 +67,7 @@ class Tuner(base_tuner.BaseTuner): by this Tuner. logger: Optional. Instance of Logger class, used for streaming data to Cloud Service for monitoring. + tuner_id: Optional. If set, use this value as the id of this Tuner. overwrite: Bool, default `False`. If `False`, reloads an existing project of the same name if one is found. Otherwise, overwrites the project. """ @@ -115,6 +116,8 @@ def __init__(self, # Save only the last N checkpoints. self._save_n_checkpoints = 10 + self.tuner_id = tuner_id or self.tuner_id + def run_trial(self, trial, *fit_args, **fit_kwargs): """Evaluates a set of hyperparameter values.