Skip to content

Commit 1169da5

Browse files
authored
fix: docstrings
1 parent 0a3884f commit 1169da5

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

bayes_opt/acquisition.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -786,6 +786,7 @@ class ExpectedImprovement(AcquisitionFunction):
786786
Decay rate for xi. If None, no decay is applied.
787787
788788
exploration_decay_delay : int, default None
789+
Delay for decay. If None, decay is applied from the start.
789790
790791
random_state : int, RandomState, default None
791792
Set the random state for reproducibility.

bayes_opt/bayesian_optimization.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,11 @@ class BayesianOptimization:
5555
Dictionary with parameters names as keys and a tuple with minimum
5656
and maximum values.
5757
58+
acquisition_function: AcquisitionFunction, optional(default=None)
59+
The acquisition function to use for suggesting new points to evaluate.
60+
If None, defaults to UpperConfidenceBound for unconstrained problems
61+
and ExpectedImprovement for constrained problems.
62+
5863
constraint: NonlinearConstraint.
5964
Note that the names of arguments of the constraint function and of
6065
f need to be the same.

0 commit comments

Comments
 (0)