Skip to content

Commit

Permalink
Remove zeroing small variance values (#684)
Browse files Browse the repository at this point in the history
* remove small variance warning

* fix commenté
  • Loading branch information
Paul-Saves authored Dec 3, 2024
1 parent 8cbd4b3 commit 571c842
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions smt/surrogate_models/krg_based.py
Original file line number Diff line number Diff line change
Expand Up @@ -1855,8 +1855,6 @@ def _predict_variances(
else:
A = self.optimal_par["sigma2"]
B = 1.0 - (rt**2.0).sum(axis=0) + (u**2.0).sum(axis=0)
# machine precision: force to zero!
B[B < 1e-12] = 0
s2 = np.einsum("i,j -> ji", A, B)
# Mean Squared Error might be slightly negative depending on
# machine precision: force to zero!
Expand Down

0 comments on commit 571c842

Please sign in to comment.