Merge pull request dongwookim-ml#3 from judaschrist/master

Error correction in gibbs sampling for sLDA

Author: dongwookim-ml

ptm/slda_gibbs.py

@@ -76,10 +76,11 @@ def stochasticEM(self, max_iter, docs, responses):
                     self.doc_topic_sum[di,old_topic] -= 1
 
                     z_bar = np.zeros([self.K,self.K]) + self.doc_topic_sum[di,:] + np.identity(self.K)
-                    z_bar /= z_bar.sum(0)
+                    # this seems more straightforward than z_bar/z_bar.sum(1)
+                    z_bar /= self.doc_topic_sum[di,:].sum() + 1
 
                     #update
-                    prob = (self.WK[word, :])/(self.sumK[:]) * (self.doc_topic_sum[di,:]) * np.exp(np.negative((responses[di] - np.dot(z_bar.T,self.eta))**2))
+                    prob = (self.WK[word, :])/(self.sumK[:]) * (self.doc_topic_sum[di,:]) * np.exp(np.negative((responses[di] - np.dot(z_bar,self.eta))**2)/2/self.sigma)
 
                     new_topic = sampling_from_dist(prob)