Mercurial Hosting > traffic-intelligence
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])