Skip to content

Commit

Permalink
Update format
Browse files Browse the repository at this point in the history
  • Loading branch information
LukasFehring committed Jan 13, 2025
1 parent 3ebdaf3 commit bd72781
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions smac/acquisition/function/prior_acquisition_function.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def _update(self, **kwargs: Any) -> None:
# Compute intiial design size
if self._initial_design_size is None:
self._initial_design_size = kwargs["num_data"]

self._iteration_number = kwargs["num_data"] - self._initial_design_size
self._eta = kwargs["eta"]

Expand Down Expand Up @@ -154,7 +154,9 @@ def _compute_prior(self, X: np.ndarray) -> np.ndarray:
for parameter, X_col in zip(self._hyperparameters.values(), X.T):
if self._discretize and isinstance(parameter, FloatHyperparameter):
assert self._discrete_bins_factor is not None
number_of_bins = int(np.ceil(self._discrete_bins_factor * self._decay_beta / (self._iteration_number + 1)))
number_of_bins = int(
np.ceil(self._discrete_bins_factor * self._decay_beta / (self._iteration_number + 1))
)
prior_values *= self._compute_discretized_pdf(parameter, X_col, number_of_bins)
else:
prior_values *= parameter._pdf(X_col[:, np.newaxis])
Expand Down

0 comments on commit bd72781

Please sign in to comment.