Mercurial Hosting > traffic-intelligence

diff scripts/learn-poi.py @ 871:6db83beb5350

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
work in progress to update gaussian mixtures
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 03 Feb 2017 16:26:18 -0500
parents 5a68779d7777
children 1cd878812529
line wrap: on
line diff
--- a/scripts/learn-poi.py	Fri Feb 03 16:15:06 2017 -0500
+++ b/scripts/learn-poi.py	Fri Feb 03 16:26:18 2017 -0500
@@ -40,7 +40,7 @@
                                    [beginnings, ends],
                                    ['beginning', 'end']):
     # estimation
-    gmm = mixture.GMM(n_components=nClusters, covariance_type = args.covarianceType)
+    gmm = mixture.GaussianMixture(n_components=nClusters, covariance_type = args.covarianceType)
     model=gmm.fit(beginnings)
     if not model.converged_:
         print('Warning: model for '+gmmType+' points did not converge')