Niko/ActiveBOToytask
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added more Acquisition functions

Browse Source
This commit is contained in:
Niko Feith 2023-02-13 13:15:08 +01:00
parent 78eccba14b
commit f54fbb7606
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
BayesianOptimization/BOwithGym.py
View File

@ -114,7 +114,7 @@ class BayesianOptimization:
self.X, self.X,
self.nr_test, self.nr_test,
self.nr_policy_weights, self.nr_policy_weights,
kappa=2.576, kappa=0,
seed=self.policy_seed, seed=self.policy_seed,
lower=self.lowerb, lower=self.lowerb,
upper=self.upperb) upper=self.upperb)
@ -191,7 +191,7 @@ class BayesianOptimization:
def main(): def main():
nr_steps = 100 nr_steps = 100
env = Continuous_MountainCarEnv() # render_mode='human' env = Continuous_MountainCarEnv() # render_mode='human'
bo = BayesianOptimization(env, nr_steps, acq='cb') bo = BayesianOptimization(env, nr_steps, acq='ei')
bo.initialize() bo.initialize()
iteration_steps = 200 iteration_steps = 200
for i in range(iteration_steps): for i in range(iteration_steps):