bayesian-optimization 3.1.0__tar.gz → 3.2.0__tar.gz

{bayesian_optimization-3.1.0 → bayesian_optimization-3.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bayesian-optimization
-Version: 3.1.0
+Version: 3.2.0
 Summary: Bayesian Optimization package
 Author: Fernando Nogueira
 Author-email: Fernando Nogueira <fmfnogueira@gmail.com>
@@ -21,9 +21,11 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: numpy>=1.25 ; python_full_version < '3.13'
 Requires-Dist: numpy>=2.1.3 ; python_full_version >= '3.13'
+Requires-Dist: packaging>=20.0
 Requires-Dist: scikit-learn>=1.0.0
 Requires-Dist: scipy>=1.0.0 ; python_full_version < '3.13'
 Requires-Dist: scipy>=1.14.1 ; python_full_version >= '3.13'

{bayesian_optimization-3.1.0 → bayesian_optimization-3.2.0}/bayes_opt/bayesian_optimization.py

@@ -8,6 +8,7 @@ from __future__ import annotations
 
 import json
 from collections import deque
+from collections.abc import Iterable
 from os import PathLike
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
@@ -175,6 +176,92 @@ class BayesianOptimization:
         """
         return self._space.res()
 
+    def predict(
+        self,
+        params: dict[str, Any] | Iterable[dict[str, Any]],
+        return_std=False,
+        return_cov=False,
+        fit_gp=True,
+    ) -> float | NDArray[Float] | tuple[float | NDArray[Float], float | NDArray[Float]]:
+        """Predict the target function value at given parameters.
+
+        Parameters
+        ---------
+        params: dict or iterable of dicts
+            The parameters where the prediction is made.
+
+        return_std: bool, optional(default=False)
+            If True, the standard deviation of the prediction is returned.
+
+        return_cov: bool, optional(default=False)
+            If True, the covariance of the prediction is returned.
+
+        fit_gp: bool, optional(default=True)
+            If True, the internal Gaussian Process model is fitted before
+            making the prediction.
+
+        Returns
+        -------
+        mean: float or np.ndarray
+            The predicted mean of the target function at the given parameters.
+            When params is a dict, returns a scalar. When params is an iterable,
+            returns a 1D array.
+
+        std_or_cov: float or np.ndarray (only if return_std or return_cov is True)
+            The predicted standard deviation or covariance of the target function
+            at the given parameters.
+        """
+        # Validate param types
+        if isinstance(params, dict):
+            params_array = self._space.params_to_array(params).reshape(1, -1)
+            single_param = True
+        elif isinstance(params, Iterable) and not isinstance(params, str):
+            # convert iterable of dicts to 2D array
+            params_array = np.array([self._space.params_to_array(p) for p in params])
+            single_param = False
+        else:
+            msg = f"params must be a dict or iterable of dicts, got {type(params).__name__}"
+            raise TypeError(msg)
+
+        # Validate mutual exclusivity of return_std and return_cov
+        if return_std and return_cov:
+            msg = "return_std and return_cov cannot both be True"
+            raise ValueError(msg)
+
+        if fit_gp:
+            if len(self._space) == 0:
+                msg = (
+                    "The Gaussian Process model cannot be fitted with zero observations. To use predict(), "
+                    "without fitting the GP, set fit_gp=False. The predictions will then be made using the "
+                    "GP prior."
+                )
+                raise RuntimeError(msg)
+            self.acquisition_function._fit_gp(self._gp, self._space)
+
+        res = self._gp.predict(params_array, return_std=return_std, return_cov=return_cov)
+
+        if return_std or return_cov:
+            mean, std_or_cov = res
+        else:
+            mean = res
+
+        # Shape semantics: dict input returns scalars, list input returns arrays
+        # Ensure list input always returns arrays (convert scalar to 1D if needed)
+        if not single_param and mean.ndim == 0:
+            mean = np.atleast_1d(mean)
+        # ruff complains when nesting conditionals, so this three-way split is necessary
+        if not single_param and (return_std or return_cov) and std_or_cov.ndim == 0:
+            std_or_cov = np.atleast_1d(std_or_cov)
+
+        if single_param and mean.ndim > 0:
+            mean = mean[0]
+        if single_param and return_std and std_or_cov.ndim > 0:
+            std_or_cov = std_or_cov[0]
+
+        if return_std or return_cov:
+            return mean, std_or_cov
+        return mean
+
     def register(
         self, params: ParamsType, target: float, constraint_value: float | NDArray[Float] | None = None
     ) -> None:
@@ -303,8 +390,8 @@ class BayesianOptimization:
             probe based on the acquisition function. This means that the GP may
             not be fitted on all points registered to the target space when the
             method completes. If you intend to use the GP model after the
-            optimization routine, make sure to fit it manually, e.g. by calling
-            ``optimizer._gp.fit(optimizer.space.params, optimizer.space.target)``.
+            optimization routine, make sure to call predict() with fit_gp=True.
+
         """
         # Log optimization start
         self.logger.log_optimization_start(self._space.keys)

{bayesian_optimization-3.1.0 → bayesian_optimization-3.2.0}/bayes_opt/logger.py

@@ -82,7 +82,7 @@ class ScreenLogger:
             result = ""
             width = self._default_cell_size
             # Keep negative sign, exponent, and as many decimal places as possible
-            if "-" in s:
+            if x < 0:
                 result += "-"
                 width -= 1
                 s = s[1:]
@@ -96,8 +96,6 @@ class ScreenLogger:
                 width -= dot_pos
                 if width > 0:
                     result += s[dot_pos : dot_pos + width]
-            else:
-                result += s[:width]
             if "e" in s:
                 result += end
             result = result.ljust(self._default_cell_size)

{bayesian_optimization-3.1.0 → bayesian_optimization-3.2.0}/bayes_opt/parameter.py

@@ -489,7 +489,10 @@ def wrap_kernel(kernel: kernels.Kernel, transform: Callable[[Any], Any]) -> kern
         def __reduce__(self) -> str | tuple[Any, ...]:
             return (wrap_kernel, (kernel, transform))
 
-    return WrappedKernel(**kernel.get_params())
+    wrapped_instance = WrappedKernel.__new__(WrappedKernel)
+    wrapped_instance.__dict__.update(kernel.__dict__)
+    wrapped_instance._transform = transform
+    return wrapped_instance
 
 
 def _copy_signature(source_fct: Callable[..., Any]) -> Callable[[Callable[..., Any]], Callable[..., Any]]:

{bayesian_optimization-3.1.0 → bayesian_optimization-3.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bayesian-optimization"
-version = "3.1.0"
+version = "3.2.0"
 description = "Bayesian Optimization package"
 authors = [{ name = "Fernando Nogueira", email = "fmfnogueira@gmail.com" }]
 license = { file = "LICENSE" }
@@ -15,11 +15,13 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
 ]
 dependencies = [
     "colorama>=0.4.6",
     "numpy>=1.25; python_version<'3.13'",
     "numpy>=2.1.3; python_version>='3.13'",
+    "packaging>=20.0",
     "scikit-learn>=1.0.0",
     "scipy>=1.0.0; python_version<'3.13'",
     "scipy>=1.14.1; python_version>='3.13'",