FIX gmm sample fix

jmschrei · Apr 2, 2016 · ccca710 · ccca710
1 parent 395ff33
commit ccca710
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pomegranate/gmm.pyx b/pomegranate/gmm.pyx
@@ -311,7 +311,7 @@ cdef class GeneralMixtureModel( Distribution ):
 
 		samples = []
 		for i in xrange(n):
-			d = numpy.random.choice( self.distributions, p=numpy.exp(self.weights), size=n )
+			d = numpy.random.choice( self.distributions, p=numpy.exp(self.weights) )
 			samples.append(d.sample())
 
 		return samples if n > 1 else samples[0]