[mlpack-svn] r16166 - mlpack/conf/jenkins-conf/benchmark/methods/scikit

fastlab-svn at coffeetalk-1.cc.gatech.edu fastlab-svn at coffeetalk-1.cc.gatech.edu
Fri Jan 24 12:50:04 EST 2014


Author: birm
Date: Fri Jan 24 12:50:03 2014
New Revision: 16166

Log:
changed scikit parameters to match mlpack defaults for benchmark. RWB

Modified:
   mlpack/conf/jenkins-conf/benchmark/methods/scikit/gmm.py

Modified: mlpack/conf/jenkins-conf/benchmark/methods/scikit/gmm.py
==============================================================================
--- mlpack/conf/jenkins-conf/benchmark/methods/scikit/gmm.py	(original)
+++ mlpack/conf/jenkins-conf/benchmark/methods/scikit/gmm.py	Fri Jan 24 12:50:03 2014
@@ -63,9 +63,10 @@
       s = 0 if not s else int(s.group(1))
 
       try:
-        # Create the Gaussian Mixture Model.
+        # Create the Gaussian Mixture Model
+	# Some params changed to match mlpack defaults.
         model = mixture.GMM(n_components=g, covariance_type='full', 
-            random_state=s, n_iter=n)
+            random_state=s, n_iter=n, n_init=10, thresh=1e-10)
         with totalTimer:
           model.fit(dataPoints)
       except Exception as e:



More information about the mlpack-svn mailing list