[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