Skip to content

Commit ef7d652

Browse files
authored
Changes default maximization strategy to UCB
1 parent 2eb0ce2 commit ef7d652

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayes_opt/bayesian_optimization.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def set_bounds(self, new_bounds):
215215
def maximize(self,
216216
init_points=5,
217217
n_iter=25,
218-
acq='ei',
218+
acq='ucb',
219219
kappa=2.576,
220220
xi=0.0,
221221
**gp_params):
@@ -340,4 +340,4 @@ def points_to_csv(self, file_name):
340340

341341
points = np.hstack((self.X, np.expand_dims(self.Y, axis=1)))
342342
header = ', '.join(self.keys + ['target'])
343-
np.savetxt(file_name, points, header=header, delimiter=',')
343+
np.savetxt(file_name, points, header=header, delimiter=',')

0 commit comments

Comments
 (0)