search

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 = [