Skip to content

Commit

Permalink
Changed hyperparameter tuning defaults
Browse files Browse the repository at this point in the history
  • Loading branch information
fjwillemsen committed Oct 23, 2024
1 parent abaf90d commit 3be5685
Showing 1 changed file with 5 additions and 12 deletions.
17 changes: 5 additions & 12 deletions kernel_tuner/hyper.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,19 +86,12 @@ def put_if_not_present(target_dict, key, value):
return list(result_unique.values()), env

if __name__ == "__main__": # TODO remove in production
# hyperparams = {
# 'popsize': [10, 20, 30],
# 'maxiter': [50, 100, 150],
# 'w': [0.25, 0.5, 0.75],
# 'c1': [1.0, 2.0, 3.0],
# 'c2': [0.5, 1.0, 1.5]
# }
hyperparams = {
'popsize': [10],
'maxiter': [50],
'w': [0.25, 0.5],
'c1': [1.0],
'c2': [0.5]
'popsize': [10, 20, 30],
'maxiter': [50, 100, 150],
'w': [0.25, 0.5, 0.75],
'c1': [1.0, 2.0, 3.0],
'c2': [0.5, 1.0, 1.5]
}
result, env = tune_hyper_params('pso', hyperparams)
print(result)
Expand Down

0 comments on commit 3be5685

Please sign in to comment.