Skip to content

Commit

Permalink
ROLLME: "jiter" -> "jitter"
Browse files Browse the repository at this point in the history
  • Loading branch information
muxator authored and marcofavoritobi committed Sep 19, 2023
1 parent 6a8558e commit eab098e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions black_it/samplers/gaussian_process.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def __init__( # pylint: disable=too-many-arguments
)
self.optimize_restarts = optimize_restarts
self.acquisition = acquisition
self.jiter = jitter
self.jitter = jitter
self._gpmodel: Optional[GaussianProcessRegressor] = None
self._fmin: Optional[Union[np.double, float]] = None

Expand Down Expand Up @@ -136,7 +136,7 @@ def predict(self, X: NDArray[np.float64]) -> NDArray[np.float64]:
# predict mean or expected improvement on the full sample set
if self.acquisition == _AcquisitionTypes.EI.value:
# minus sign needed for subsequent sorting
candidates_score = -self._predict_EI(X, self.jiter)
candidates_score = -self._predict_EI(X, self.jitter)
else: # acquisition is "mean"
candidates_score = self._predict_mean_std(X)[0]

Expand Down

0 comments on commit eab098e

Please sign in to comment.