Skip to content

Commit a6697aa

Browse files
authored
Acquisition function API redesign (#447)
* Redesign the acquisition function API * Add async optimization w/ kriging believer example * Use `typing.Union` instead of `|` * Add ConstantLiar, fix bugs, reran code * Fix typo * Slight changes, add many tests * Add `base_acq` `ThompsonSampling`, fix a bug in `KrigingBeliever` * Clean code, add docstrings * add tests, rm kriging believer, dont use gp hedge as default * Use `self.base_acq` instead of supplying it as an argument to `._get_acq` * Add special exceptions covering too few registered points * Notebook to showcase acquisition functions * Linting, code style * Drop references to believer from script * Delete deprecated parameters from `suggest` * Use `abc.ABC` for acquisition base class * Add more tests * Fix type signature breaking Python 3.9 * Trigger CI * Clean docstring
1 parent 48e4816 commit a6697aa

16 files changed

+2096
-760
lines changed

bayes_opt/__init__.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,19 @@
11
"""Pure Python implementation of bayesian global optimization with gaussian processes."""
22
from .bayesian_optimization import BayesianOptimization, Events
3-
from .domain_reduction import SequentialDomainReductionTransformer
4-
from .util import UtilityFunction
3+
from .domain_reduction import SequentialDomainReductionTransformer
54
from .logger import ScreenLogger, JSONLogger
65
from .constraint import ConstraintModel
7-
from .util import UtilityFunction
6+
from . import acquisition
87

98
import importlib.metadata
109
__version__ = importlib.metadata.version('bayesian-optimization')
1110

1211

1312

1413
__all__ = [
14+
"acquisition",
1515
"BayesianOptimization",
1616
"ConstraintModel",
17-
"UtilityFunction",
1817
"Events",
1918
"ScreenLogger",
2019
"JSONLogger",

0 commit comments

Comments
 (0)