fixed std calculation

experimental-design · Sep 18, 2024 · 64cf2bc · 64cf2bc
1 parent 46c7000
commit 64cf2bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bofire/surrogates/multi_task_gp.py b/bofire/surrogates/multi_task_gp.py
@@ -2,6 +2,7 @@
 from typing import Dict, Optional
 
 import botorch
+import numpy as np
 import pandas as pd
 import torch
 from botorch.fit import fit_gpytorch_mll
@@ -100,7 +101,7 @@ def _predict(self, transformed_X: pd.DataFrame):
             except NotImplementedError:
                 posterior = self.model.posterior(X=X, observation_noise=False)  # type: ignore
             preds = posterior.mean.cpu().detach().numpy()  # type: ignore
-            stds = posterior.variance.cpu().detach().numpy()  # type: ignore
+            stds = np.sqrt(posterior.variance.cpu().detach().numpy())  # type: ignore
 
         return preds, stds