Skip to content

Commit c05ed93

Browse files
committed
Change how n_initial_points is passed to skopt.Optimizer
1 parent d451a2a commit c05ed93

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

kernel_tuner/strategies/skopt.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,14 +46,23 @@ def tune(searchspace: Searchspace, runner, tuning_options):
4646

4747
# Create skopt optimizer
4848
skopt_kwargs = dict(skopt_kwargs)
49-
skopt_kwargs.setdefault("acq_func_kwargs", {})["kappa"] = kappa
49+
skopt_kwargs["base_estimator"] = learner
50+
skopt_kwargs["acq_func"] = acq_func
51+
52+
# Only set n_initial_points if not None
53+
if num_initial is not None:
54+
skopt_kwargs["n_initial_points"] = num_initial
55+
56+
# Set kappa is not None
57+
if kappa is not None:
58+
skopt_kwargs.setdefault("acq_func_kwargs", {})["kappa"] = kappa
59+
60+
if tuning_options.verbose:
61+
print(f"Initialize scikit-optimize Optimizer object: {skopt_kwargs}")
5062

5163
from skopt import Optimizer as SkOptimizer
5264
optimizer = SkOptimizer(
5365
dimensions=bounds,
54-
base_estimator=learner,
55-
n_initial_points=num_initial,
56-
acq_func=acq_func,
5766
space_constraint=space_constraint,
5867
**skopt_kwargs
5968
)

0 commit comments

Comments
 (0)