From 9acc68b5c2f8daf63354bf87df78fb5cdf7d9834 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 9 Feb 2018 20:18:10 -0500
Subject: [PATCH] fewer models
---
.../models_class1/generate_hyperparameters.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/downloads-generation/models_class1/generate_hyperparameters.py b/downloads-generation/models_class1/generate_hyperparameters.py
index fc9ef8bc..c9f54646 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)
--
GitLab