From deabe4dde60d95dcdcee8f700e4e05cf44c97279 Mon Sep 17 00:00:00 2001
From: Michael Pilosov <michael.pilosov@ucdenver.edu>
Date: Sun, 14 Jan 2024 03:51:22 +0000
Subject: [PATCH] search

---
 search.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/search.py b/search.py
index 7a9b3cc..d4413aa 100644
--- a/search.py
+++ b/search.py
@@ -18,10 +18,10 @@ job_plugin = studio.installed_plugins["jobs"]
 # do a sweep over learning rates
 
 # Define the ranges or sets of values for each hyperparameter
-alpha_values = list(np.round(np.linspace(0, 2, 21), 4))
-learning_rate_values = [1e-3, 1e-4, 1e-5]
-batch_size_values = [128]
-max_epochs_values = [5000]
+alpha_values = list(np.round(np.linspace(0, 4, 41), 4))
+learning_rate_values = np.log(np.linspace(-5, -3, 11))
+batch_size_values = [64, 128]
+max_epochs_values = [500]
 
 # Generate all possible combinations of hyperparameters
 all_params = [