未验证 提交 5ab7287f 编辑于 作者: dlagul's avatar dlagul 提交者： GitHub
浏览文件

Update tester.py

上级 2af08acc

sdfvae/tester.py

+2 −0
原始行号 差异行号 差异行
@@ -87,6 +87,7 @@ class Tester(object): 
        # log(N(x|mu,sigma^2))

        # = log{1/(sqrt(2*pi)*sigma)exp{-(x-mu)^2/(2*sigma^2)}} 

        # = -0.5*{log(2*pi)+2*log(sigma)+[(x-mu)/exp{log(sigma)}]^2}

        # Note that var = sigma^2, so “recon_seq_logvar” is more appropriate to be called “recon_seq_logsigma”, but the name does not the matter

        loglikelihood = -0.5 * torch.sum(torch.pow(((original_seq.float()-recon_seq_mu.float())/torch.exp(recon_seq_logvar.float())), 2) 

                                         + 2 * recon_seq_logvar.float() 

                                         + np.log(np.pi*2))
@@ -109,6 +110,7 @@ class Tester(object): 
        # log(N(x|mu,sigma^2))

        # = log{1/(sqrt(2*pi)*sigma)exp{-(x-mu)^2/(2*sigma^2)}} 

        # = -0.5*{log(2*pi)+2*log(sigma)+[(x-mu)/exp{log(sigma)}]^2}

        # Note that var = sigma^2, so “recon_seq_logvar” is more appropriate to be called “recon_seq_logsigma”, but the name does not the matter

        llh = -0.5 * torch.sum(torch.pow(((x.float()-recon_x_mu.float())/torch.exp(recon_x_logvar.float())), 2) 

              + 2 * recon_x_logvar.float() 

              + np.log(np.pi*2))