FIX icd training

jmschrei · Mar 6, 2016 · 6f2c996 · 6f2c996
1 parent 7ca9dcc
commit 6f2c996
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pomegranate/distributions.pyx b/pomegranate/distributions.pyx
@@ -394,7 +394,7 @@ cdef class NormalDistribution( Distribution ):
 
 	cdef void _summarize( self, double* items, double* weights, SIZE_t n ) nogil:
 		"""Cython function to get the MLE estimate for a Gaussian."""
-		
+
 		cdef SIZE_t i
 		cdef double x_sum = 0.0, x2_sum = 0.0, w_sum = 0.0
 
@@ -1967,7 +1967,7 @@ cdef class IndependentComponentsDistribution( MultivariateDistribution ):
 
 		for i in range(n):
 			for j in range(d):
-				( <Distribution> self.distributions_ptr[j] )._summarize( items+i*d+j, weights+i*d+j, 1 )
+				( <Distribution> self.distributions_ptr[j] )._summarize( items+i*d+j, weights+i, 1 )
 
 	def from_summaries( self, inertia=0.0 ):
 		"""

diff --git a/pomegranate/hmm.pyx b/pomegranate/hmm.pyx
@@ -2245,7 +2245,7 @@ cdef class HiddenMarkovModel( Model ):
 							weights[i] += cexp( f[(i+1)*m + li] + b[(i+1)*m + li] -
 								log_sequence_probability )
 
-					(<Distribution>distributions[k])._summarize( sequence, weights, n )
+					(<Distribution>distributions[k])._summarize(sequence, weights, n)
 
 			# Update the master expected transitions vector representing the sparse matrix.
 			with gil: