Skip to content
Snippets Groups Projects
Commit 9acc68b5 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fewer models

parent f4a49a03
No related merge requests found
Hide whitespace changes
Inline Side-by-side
downloads-generation/models_class1/generate_hyperparameters.py
+ 2
− 2
View file @ 9acc68b5
...@@ -10,7 +10,7 @@ base_hyperparameters = { ...@@ -10,7 +10,7 @@ base_hyperparameters = {
########################################## ##########################################
# ENSEMBLE SIZE # ENSEMBLE SIZE
########################################## ##########################################
"n_models": 8, "n_models": 4,
########################################## ##########################################
# OPTIMIZATION # OPTIMIZATION
...@@ -66,7 +66,7 @@ grid = [] ...@@ -66,7 +66,7 @@ grid = []
for train_subset in ["all", "quantitative"]: for train_subset in ["all", "quantitative"]:
for minibatch_size in [128]: for minibatch_size in [128]:
for dense_layer_size in [8, 16, 32, 64]: 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 num_lc in [0, 1, 2]:
for lc_kernel_size in [3, 5]: for lc_kernel_size in [3, 5]:
new = deepcopy(base_hyperparameters) new = deepcopy(base_hyperparameters)
......
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