From c2d81f115954b8683199076857b1d7184290a0df Mon Sep 17 00:00:00 2001 From: Max Burg Date: Mon, 30 May 2022 12:43:51 +0200 Subject: [PATCH] [add] docstring for new seed_config_auto argument --- nnfabrik/utility/hypersearch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nnfabrik/utility/hypersearch.py b/nnfabrik/utility/hypersearch.py index 4efe74a..4a00028 100644 --- a/nnfabrik/utility/hypersearch.py +++ b/nnfabrik/utility/hypersearch.py @@ -266,6 +266,8 @@ class Random: trainer_fn (str): name of the trainer function trainer_config (dict): dictionary of arguments for trainer function that are fixed trainer_config_auto (dict): dictionary of arguments for trainer function that are to be randomly sampled + seed_config_auto (dict): dictionary of arguments for setting (`dict(seed={"type": "fixed", "value": })`) + or random sampling (`dict(seed={"type": "int"})`) the seed architect (str): Name of the contributor that added this entry trained_model_table (str): name (importable) of the trained_model_table total_trials (int, optional): Number of experiments (i.e. training) to run. Defaults to 5. @@ -499,7 +501,5 @@ def run(self): """ Runs the random hyperparameter search, for as many trials as specified. """ - # n_trials = len(self.trained_model_table().seed_table()) * self.total_trials - # init_len = len(self.trained_model_table()) for _ in range(self.total_trials): self.train_evaluate(self.gen_params_value())