Skip to content

GitLab

Unverified Commit e42c769c authored by Niru Maheswaranathan's avatar Niru Maheswaranathan Committed by GitHub
Browse files

[LFADS] Fixes typo in distributions.py (#6161)

parent d6b2b83c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/lfads/distributions.py
View file @ e42c769c
...@@ -391,7 +391,7 @@ class LearnableAutoRegressive1Prior(GaussianProcess): ...@@ -391,7 +391,7 @@ class LearnableAutoRegressive1Prior(GaussianProcess):
Returns: Returns:
The likelihood of p_t under the model at time t. i.e. The likelihood of p_t under the model at time t. i.e.
p(z_t|z_tm1) = N(z_tm1 * phis, eps^2) p(z_t|z_tm1_bxu) = N(z_tm1_bxu * phis, eps^2)
""" """
if z_tm1_bxu is None: if z_tm1_bxu is None:
...@@ -485,7 +485,7 @@ class KLCost_GaussianGaussianProcessSampled(object): ...@@ -485,7 +485,7 @@ class KLCost_GaussianGaussianProcessSampled(object):
z_t_bxu = z_t.sample z_t_bxu = z_t.sample
logq_bxu += z_t.logp(z_t_bxu) logq_bxu += z_t.logp(z_t_bxu)
logp_bxu += prior_z_process.logp_t(z_t_bxu, z_tm1_bxu) logp_bxu += prior_z_process.logp_t(z_t_bxu, z_tm1_bxu)
z_tm1 = z_t_bxu z_tm1_bxu = z_t_bxu
kl_bxu = logq_bxu - logp_bxu kl_bxu = logq_bxu - logp_bxu
kl_b = tf.reduce_sum(kl_bxu, [1]) kl_b = tf.reduce_sum(kl_bxu, [1])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment