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

fixed bo service

Browse Source
This commit is contained in:
Niko Feith 2023-03-07 17:12:21 +01:00
parent a05e5af477
commit 33764e1ac3
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/active_bo_ros/active_bo_ros/BayesianOptimization/BayesianOptimization.py
View File

@ -3,9 +3,9 @@ from sklearn.gaussian_process import GaussianProcessRegressor
from sklearn.gaussian_process.kernels import Matern from sklearn.gaussian_process.kernels import Matern
from active_bo_ros.PolicyModel.GaussianRBFModel import GaussianRBF from active_bo_ros.PolicyModel.GaussianRBFModel import GaussianRBF
from active_bo_ros.AcquistionFunctions.ExpectedImprovement import ExpectedImprovement from active_bo_ros.AcquisitionFunctions.ExpectedImprovement import ExpectedImprovement
from active_bo_ros.AcquistionFunctions.ProbabilityOfImprovement import ProbabilityOfImprovement from active_bo_ros.AcquisitionFunctions.ProbabilityOfImprovement import ProbabilityOfImprovement
from active_bo_ros.AcquistionFunctions.ConfidenceBound import ConfidenceBound from active_bo_ros.AcquisitionFunctions.ConfidenceBound import ConfidenceBound
class BayesianOptimization: class BayesianOptimization:
def __init__(self, env, nr_steps, nr_init=3, acq='ei', nr_weights=6, policy_seed=None): def __init__(self, env, nr_steps, nr_init=3, acq='ei', nr_weights=6, policy_seed=None):

2
src/active_bo_ros/setup.py
View File

@ -18,7 +18,7 @@ setup(
('share/' + package_name, ['package.xml']), ('share/' + package_name, ['package.xml']),
(os.path.join('share', package_name), glob('launch/*.launch.py')), (os.path.join('share', package_name), glob('launch/*.launch.py')),
], ],
install_requires=['setuptools', 'gym', 'numpy', 'sklearn'], install_requires=['setuptools', 'gym', 'numpy'],
zip_safe=True, zip_safe=True,
maintainer='cpsfeith', maintainer='cpsfeith',
maintainer_email='nikolaus.feith@unileoben.ac.at', maintainer_email='nikolaus.feith@unileoben.ac.at',