Skip to content

Commit

Permalink
Reactivate normalize_y and improve hidden constant values
Browse files Browse the repository at this point in the history
These parameters will soon be converted into CLI parameters such that
the user is able to set them. Until then they will be set to more plausible
values.
  • Loading branch information
kiudee committed Mar 22, 2021
1 parent a41079b commit a905bd4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tune/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -281,7 +281,7 @@ def local( # noqa: C901
random_state = np.random.RandomState(np.random.MT19937(ss.spawn(1)[0]))
gp_kwargs = dict(
# TODO: Due to a bug in scikit-learn 0.23.2, we set normalize_y=False:
normalize_y=False,
normalize_y=True,
warp_inputs=settings.get("warp_inputs", warp_inputs),
)
opt = Optimizer(
Expand All @@ -292,7 +292,7 @@ def local( # noqa: C901
gp_kwargs=gp_kwargs,
# gp_priors=priors, # TODO: Let user pass in priors
acq_func=settings.get("acq_function", acq_function),
acq_func_kwargs=dict(alpha="inf", n_thompson=20),
acq_func_kwargs=dict(alpha=1.96, n_thompson=500),
random_state=random_state,
)
X = []
Expand Down

0 comments on commit a905bd4

Please sign in to comment.