Update pf_multiSample.py

hansenlab · Oct 9, 2024 · 9de9f07 · 9de9f07
1 parent 7892dac
commit 9de9f07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mNSF/pf_multiSample.py b/mNSF/pf_multiSample.py
@@ -341,7 +341,7 @@ def elbo_avg(self, X, Y, sz=1, S=1, Ntot=None, chol=True):
  #kl_terms is not affected by minibatching so use reduce_sum
  #print(1111)
  kl_term = tf.reduce_sum(self.eval_kl_term(mu_z, Kuu_chol))
- Mu = self.sample_predictive_mean(X, sz=sz, S=S, kernel=ker, mu_z=mu_z, Kuu_chol=Kuu_chol)
+ Mu = self.sample_predictive_mean(X, sz=sz, S=S, kernel=ker, mu_z=mu_z, Kuu_chol=Kuu_chol, chol = chol)
  eloglik = likelihoods.lik_to_distr(self.lik, Mu, self.disp).log_prob(Y)
  return J*tf.reduce_mean(eloglik) - kl_term/Ntot