Skip to content

Commit

Permalink
fix elbo
Browse files Browse the repository at this point in the history
  • Loading branch information
ferrine authored and twiecki committed Sep 24, 2021
1 parent 2257b73 commit 4eb12f3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pymc3/variational/approximations.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ def symbolic_logq_not_scaled(self):
z0 = self.symbolic_initial
std = rho2sigma(self.rho)
logdet = at.log(std)
quaddist = ((z0) ** 2 + at.log(np.pi / 2.0)) / 2.0
quaddist = -0.5 * z0 ** 2 - at.log((2 * np.pi) ** 0.5)
logq = quaddist - logdet
return logq.sum(range(1, logq.ndim))

Expand Down

0 comments on commit 4eb12f3

Please sign in to comment.