Fixed two small lbugs

This commit is contained in:
Siivola Eero 2017-07-04 13:18:06 +03:00 committed by Akash Kumar Dhaka
parent 5d7e575c77
commit a9cdfad4b6
2 changed files with 2 additions and 2 deletions

View file

@ -175,7 +175,7 @@ class EP(EPBase, ExactGaussianInference):
if self.ep_mode=="nested":
#Force EP at each step of the optimization
self._ep_approximation = None
post_params, ga_approx, log_Z_tilde = self._ep_approximation = self.expectation_propagation(K, Y, likelihood, Y_metadata)
post_params, ga_approx, cav_params, log_Z_tilde = self._ep_approximation = self.expectation_propagation(K, Y, likelihood, Y_metadata)
elif self.ep_mode=="alternated":
if getattr(self, '_ep_approximation', None) is None:
#if we don't yet have the results of runnign EP, run EP and store the computed factors in self._ep_approximation