diff --git a/downloads-generation/models_class1/generate_hyperparameters.py b/downloads-generation/models_class1/generate_hyperparameters.py index e958e101639acc5bc1a4524d6f73de22e3528a98..579e88da06581b5c5bfca9510b3093ebc80bd698 100644 --- a/downloads-generation/models_class1/generate_hyperparameters.py +++ b/downloads-generation/models_class1/generate_hyperparameters.py @@ -70,6 +70,7 @@ for train_subset in ["all", "quantitative"]: for num_lc in [0, 1, 2]: for lc_kernel_size in [3, 5]: new = deepcopy(base_hyperparameters) + new["minibatch_size"] = minibatch_size new["train_data"]["subset"] = train_subset new["layer_sizes"] = [dense_layer_size] new["dense_layer_l1_regularization"] = l1