From f54fbb760654f4d0970eae1ea3c06dc899be0233 Mon Sep 17 00:00:00 2001 From: "nikolaus.feith" Date: Mon, 13 Feb 2023 13:15:08 +0100 Subject: [PATCH] added more Acquisition functions --- BayesianOptimization/BOwithGym.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/BayesianOptimization/BOwithGym.py b/BayesianOptimization/BOwithGym.py index 025f142..1aa85ad 100644 --- a/BayesianOptimization/BOwithGym.py +++ b/BayesianOptimization/BOwithGym.py @@ -114,7 +114,7 @@ class BayesianOptimization: self.X, self.nr_test, self.nr_policy_weights, - kappa=2.576, + kappa=0, seed=self.policy_seed, lower=self.lowerb, upper=self.upperb) @@ -191,7 +191,7 @@ class BayesianOptimization: def main(): nr_steps = 100 env = Continuous_MountainCarEnv() # render_mode='human' - bo = BayesianOptimization(env, nr_steps, acq='cb') + bo = BayesianOptimization(env, nr_steps, acq='ei') bo.initialize() iteration_steps = 200 for i in range(iteration_steps):