added more Acquisition functions

2023-02-13 13:15:08 +01:00 · 2023-02-13 13:15:08 +01:00 · f54fbb7606
commit f54fbb7606
parent 78eccba14b
1 changed files with 2 additions and 2 deletions
--- 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):