Skip to content

Commit b68473f

Browse files
committed
adding method to set nugget
1 parent 1db527a commit b68473f

File tree

1 file changed

+4
-0
lines changed
  • src/hiopbbpy/surrogate_modeling

1 file changed

+4
-0
lines changed

src/hiopbbpy/surrogate_modeling/krg.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,3 +60,7 @@ def variance_gradient(self, x: np.ndarray) -> np.ndarray:
6060
if not self.trained:
6161
raise ValueError("must train kriging model before utilizing it to predict gradient")
6262
return self.surrogatesmt.predict_variance_gradient(x)
63+
64+
def set_nugget(self, nugget):
65+
assert nugget >= 0., "nugget value must be non-negative"
66+
self.surrogatesmt.options["nugget"] = nugget

0 commit comments

Comments
 (0)