changeset 21:1a2cfd98e737

remove GMM parameter cvtype diag should be fine, and param doesn't exist in all sklearn.mixture versions
author Dan Stowell <danstowell@users.sourceforge.net>
date Thu, 10 Jan 2013 14:31:40 +0000
parents 7d6a647a2d44
children c7fa1f02f5f8 54612c177bcc
files smacpy.py
diffstat 1 files changed, 1 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/smacpy.py	Thu Jan 10 10:33:30 2013 +0000
+++ b/smacpy.py	Thu Jan 10 14:31:40 2013 +0000
@@ -76,7 +76,7 @@
 		self.gmms = {}
 		for label, aggf in aggfeatures.items():
 			if verbose: print("    Training a GMM for label %s, using data of shape %s" % (label, str(np.shape(aggf))))
-			self.gmms[label] = GMM(n_components=10, cvtype='full')
+			self.gmms[label] = GMM(n_components=10) # , cvtype='full')
 			self.gmms[label].fit(aggf)
 		if verbose: print("  Trained %i classes from %i input files" % (len(self.gmms), len(trainingdata)))