sketch of how default kernel will change

experimental-design · Aug 27, 2024 · b6a78e2 · b6a78e2
1 parent 3b75089
commit b6a78e2
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/bofire/data_models/surrogates/single_task_gp.py b/bofire/data_models/surrogates/single_task_gp.py
@@ -22,6 +22,8 @@
     BOTORCH_LENGTHCALE_PRIOR,
     BOTORCH_NOISE_PRIOR,
     BOTORCH_SCALE_PRIOR,
+    HVARFNER_LENGTHSCALE_PRIOR,
+    HVARFNER_NOISE_PRIOR,
     MBO_LENGTHCALE_PRIOR,
     MBO_NOISE_PRIOR,
     MBO_OUTPUTSCALE_PRIOR,
@@ -99,16 +101,13 @@ class SingleTaskGPSurrogate(TrainableBotorchSurrogate):
     type: Literal["SingleTaskGPSurrogate"] = "SingleTaskGPSurrogate"
 
     kernel: AnyKernel = Field(
-        default_factory=lambda: ScaleKernel(
-            base_kernel=MaternKernel(
-                ard=True,
-                nu=2.5,
-                lengthscale_prior=BOTORCH_LENGTHCALE_PRIOR(),
-            ),
-            outputscale_prior=BOTORCH_SCALE_PRIOR(),
+        default_factory=lambda: MaternKernel(
+            ard=True,
+            nu=2.5,
+            lengthscale_prior=HVARFNER_LENGTHSCALE_PRIOR(),
         )
     )
-    noise_prior: AnyPrior = Field(default_factory=lambda: BOTORCH_NOISE_PRIOR())
+    noise_prior: AnyPrior = Field(default_factory=lambda: HVARFNER_NOISE_PRIOR())
     hyperconfig: Optional[SingleTaskGPHyperconfig] = Field(
         default_factory=lambda: SingleTaskGPHyperconfig()
     )