diff --git a/src/active_bo_ros/active_bo_ros/BayesianOptimization/BayesianOptimization.py b/src/active_bo_ros/active_bo_ros/BayesianOptimization/BayesianOptimization.py index 6e3077d..db1e211 100644 --- a/src/active_bo_ros/active_bo_ros/BayesianOptimization/BayesianOptimization.py +++ b/src/active_bo_ros/active_bo_ros/BayesianOptimization/BayesianOptimization.py @@ -3,9 +3,9 @@ from sklearn.gaussian_process import GaussianProcessRegressor from sklearn.gaussian_process.kernels import Matern from active_bo_ros.PolicyModel.GaussianRBFModel import GaussianRBF -from active_bo_ros.AcquistionFunctions.ExpectedImprovement import ExpectedImprovement -from active_bo_ros.AcquistionFunctions.ProbabilityOfImprovement import ProbabilityOfImprovement -from active_bo_ros.AcquistionFunctions.ConfidenceBound import ConfidenceBound +from active_bo_ros.AcquisitionFunctions.ExpectedImprovement import ExpectedImprovement +from active_bo_ros.AcquisitionFunctions.ProbabilityOfImprovement import ProbabilityOfImprovement +from active_bo_ros.AcquisitionFunctions.ConfidenceBound import ConfidenceBound class BayesianOptimization: def __init__(self, env, nr_steps, nr_init=3, acq='ei', nr_weights=6, policy_seed=None): diff --git a/src/active_bo_ros/setup.py b/src/active_bo_ros/setup.py index be8e6d3..98587c5 100644 --- a/src/active_bo_ros/setup.py +++ b/src/active_bo_ros/setup.py @@ -18,7 +18,7 @@ setup( ('share/' + package_name, ['package.xml']), (os.path.join('share', package_name), glob('launch/*.launch.py')), ], - install_requires=['setuptools', 'gym', 'numpy', 'sklearn'], + install_requires=['setuptools', 'gym', 'numpy'], zip_safe=True, maintainer='cpsfeith', maintainer_email='nikolaus.feith@unileoben.ac.at',