From f4a49a0321d12e58b59dd630f8ce0acced9a06fd Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 9 Feb 2018 20:16:36 -0500
Subject: [PATCH] search fewer hyperparameters
---
downloads-generation/models_class1/generate_hyperparameters.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/downloads-generation/models_class1/generate_hyperparameters.py b/downloads-generation/models_class1/generate_hyperparameters.py
index 45767879..fc9ef8bc 100644
--- a/downloads-generation/models_class1/generate_hyperparameters.py
+++ b/downloads-generation/models_class1/generate_hyperparameters.py
@@ -64,7 +64,7 @@ base_hyperparameters = {
grid = []
for train_subset in ["all", "quantitative"]:
- for minibatch_size in [128, 512]:
+ for minibatch_size in [128]:
for dense_layer_size in [8, 16, 32, 64]:
for l1 in [0.0, 0.001, 0.01]:
for num_lc in [0, 1, 2]:
--
GitLab