diff --git a/downloads-generation/models_class1/generate_hyperparameters.py b/downloads-generation/models_class1/generate_hyperparameters.py index fc9ef8bcf72f05802f28eca570a117434ef5f6fa..c9f546460060f9d3beab947cfe5a394ee3e79e9c 100644 --- a/downloads-generation/models_class1/generate_hyperparameters.py +++ b/downloads-generation/models_class1/generate_hyperparameters.py @@ -10,7 +10,7 @@ base_hyperparameters = { ########################################## # ENSEMBLE SIZE ########################################## - "n_models": 8, + "n_models": 4, ########################################## # OPTIMIZATION @@ -66,7 +66,7 @@ grid = [] for train_subset in ["all", "quantitative"]: for minibatch_size in [128]: for dense_layer_size in [8, 16, 32, 64]: - for l1 in [0.0, 0.001, 0.01]: + for l1 in [0.0, 0.001]: for num_lc in [0, 1, 2]: for lc_kernel_size in [3, 5]: new = deepcopy(base_hyperparameters)