Skip to content

Commit b4acb32

Browse files
committed
revert ELBO to formulation on zeta rather than z
1 parent 943e60b commit b4acb32

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/elbo.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,11 @@ function neg_elbo_transnorm_gf(rng, g, f, ϕ::AbstractVector, y_ob, x::AbstractM
3131
nLy1 = neg_logden_indep_normal(y_ob, y_pred_i, logσ2y)
3232
nLy1 - logjac
3333
end) / n_MC
34-
logdet_jacT2 = sum_log_σ = sum(log.(σ))
34+
#sum_log_σ = sum(log.(σ))
3535
# logdet_jacT2 = -sum_log_σ # log Prod(1/σ_i) = -sum log σ_i
36-
# logdetΣ = 2 * sum_log_σ # log Prod(σ_i²) = 2* sum log σ_i
37-
# ent = entropy_MvNormal(size(ζs, 1), logdetΣ) # defined in logden_normal
38-
nLy - logdet_jacT2 - entropyN
36+
logdetΣ = 2 * sum(log.(σ))
37+
entropy_ζ = entropy_MvNormal(size(ζs, 1), logdetΣ) # defined in logden_normal
38+
nLy - entropy_ζ
3939
end
4040

4141
"""

0 commit comments

Comments
 (0)