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