fix loc and scale_diag kwargs to ParamMixture in unsupervised tutorial

blei-lab · Apr 27, 2017 · 52fef60 · 52fef60
1 parent 39b6cea
commit 52fef60
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 41 deletions.
diff --git a/docs/tex/tutorials/unsupervised.tex b/docs/tex/tutorials/unsupervised.tex
@@ -106,7 +106,7 @@ \subsubsection{Model}
 pi = Dirichlet(tf.ones(K))
 mu = Normal(tf.zeros(D), tf.ones(D), sample_shape=K)
 sigmasq = InverseGamma(tf.ones(D), tf.ones(D), sample_shape=K)
-x = ParamMixture(pi, {'mu': mu, 'diag_stdev': tf.sqrt(sigmasq)}, MultivariateNormalDiag,
+x = ParamMixture(pi, {'loc': mu, 'scale_diag': tf.sqrt(sigmasq)}, MultivariateNormalDiag,
                  sample_shape=N)
 z = x.cat
 \end{lstlisting}

diff --git a/edward/inferences/conjugacy/conjugate_log_probs.py b/edward/inferences/conjugacy/conjugate_log_probs.py
@@ -122,7 +122,7 @@ def multinomial_log_prob(self, val):
 @_val_wrapper
 def mvn_diag_log_prob(self, val):
   loc = self.parameters['loc']
-  scale_diag = self.parameters['scale'].diag
+  scale_diag = self.parameters['scale_diag']
   prec = tf.reciprocal(tf.square(scale_diag))
   result = prec * (-0.5 * tf.square(val) - 0.5 * tf.square(loc) +
                    val * loc)