diff trafficintelligence/storage.py @ 1204:a12d126346ff

merge
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 22 Mar 2023 22:01:12 -0400
parents 7b3384a8e409 4356065ed3ca
children 3905b393ade0
line wrap: on
line diff
--- a/trafficintelligence/storage.py	Wed Mar 22 15:40:33 2023 -0400
+++ b/trafficintelligence/storage.py	Wed Mar 22 22:01:12 2023 -0400
@@ -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])