Mercurial Hosting > traffic-intelligence

diff trafficintelligence/storage.py @ 1200:4356065ed3ca

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
updated simple moving average filter and cleaned tests
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 08 Dec 2022 10:24:22 -0500
parents ccab20f85710
children a12d126346ff
line wrap: on
line diff
--- a/trafficintelligence/storage.py	Fri Nov 18 15:55:09 2022 -0500
+++ b/trafficintelligence/storage.py	Thu Dec 08 10:24:22 2022 -0500
@@ -756,7 +756,7 @@
                                     'weight': row[7],
                                     'precisions': array(literal_eval(row[8]))})
                 if len(gmm) > 0:
-                    tmp = mixture.GaussianMixture(len(gmm), covarianceType)
+                    tmp = mixture.GaussianMixture(n_components=len(gmm), covariance_type=covarianceType)
                     tmp.means_ = array([gaussian['mean'] for gaussian in gmm])
                     tmp.covariances_ = array([gaussian['covar'] for gaussian in gmm])
                     tmp.weights_ = array([gaussian['weight'] for gaussian in gmm])