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):