Skip to content

Commit 3be5685

Browse files
committed
Changed hyperparameter tuning defaults
1 parent abaf90d commit 3be5685

File tree

1 file changed

+5
-12
lines changed

1 file changed

+5
-12
lines changed

kernel_tuner/hyper.py

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -86,19 +86,12 @@ def put_if_not_present(target_dict, key, value):
8686
return list(result_unique.values()), env
8787

8888
if __name__ == "__main__": # TODO remove in production
89-
# hyperparams = {
90-
# 'popsize': [10, 20, 30],
91-
# 'maxiter': [50, 100, 150],
92-
# 'w': [0.25, 0.5, 0.75],
93-
# 'c1': [1.0, 2.0, 3.0],
94-
# 'c2': [0.5, 1.0, 1.5]
95-
# }
9689
hyperparams = {
97-
'popsize': [10],
98-
'maxiter': [50],
99-
'w': [0.25, 0.5],
100-
'c1': [1.0],
101-
'c2': [0.5]
90+
'popsize': [10, 20, 30],
91+
'maxiter': [50, 100, 150],
92+
'w': [0.25, 0.5, 0.75],
93+
'c1': [1.0, 2.0, 3.0],
94+
'c2': [0.5, 1.0, 1.5]
10295
}
10396
result, env = tune_hyper_params('pso', hyperparams)
10497
print(result)

0 commit comments

Comments
 (0)