Skip to content

Commit 9f77f5c

Browse files
authored
Avoid diagonal matrix creation in FITC (#499)
1 parent 37bb5c1 commit 9f77f5c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

smt/surrogate_models/sgp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ def _fitc(self, X, Y, Z, theta, sigma2, noise, nugget):
233233
beta = 1.0 / nu
234234

235235
# Compute (lower) Cholesky decomposition: A = I + V diag(beta) V^T = L L^T
236-
A = np.eye(self.nz) + V @ np.diag(beta) @ V.T
236+
A = np.eye(self.nz) + V * beta @ V.T
237237
L = linalg.cholesky(A, lower=True)
238238
Li = linalg.inv(L)
239239

0 commit comments

Comments
 (0)