cobyla for sgp

Paul-Saves · Jan 29, 2024 · 11811dd · 11811dd
1 parent 9b71a3a
commit 11811dd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/smt/surrogate_models/sgp.py b/smt/surrogate_models/sgp.py
@@ -50,6 +50,13 @@ def _initialize(self):
             desc="Gaussian noise on observed training data",
             types=(list, np.ndarray),
         )
+        declare(
+            "hyper_opt",
+            "Cobyla",
+            values=("Cobyla"),
+            desc="Optimiser for hyperparameters optimisation",
+            types=str,
+        )
         declare(
             "eval_noise",
             True,  # for SGP evaluate noise by default