Update pf.py

hansenlab · Oct 9, 2024 · 9923f1e · 9923f1e
1 parent adf250b
commit 9923f1e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mNSF/NSF/pf.py b/mNSF/NSF/pf.py
@@ -99,6 +99,7 @@ def __init__(self, J, L, Z, lik="poi", chol = True, X=None, psd_kernel=tfk.Mater
  self._init_misc()
  self.Kuu_chol = tf.Variable(self.eval_Kuu_chol(self.get_kernel()), dtype=dtp, trainable=False)
  if not(chol): 
+ kernel = self.kernel 
  N = X.shape[0]
  mu_x = self.beta0+tfl.matmul(self.beta, X, transpose_b=True) #LxN
  self.Kuf = kernel.matrix(self.Z, X) #LxMxN