Mercurial Hosting > traffic-intelligence

comparison 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
comparison
equal deleted inserted replaced
1199:6a6a4d5958f7 1200:4356065ed3ca
754 'mean': row[3:5], 754 'mean': row[3:5],
755 'covar': array(literal_eval(row[5])), 755 'covar': array(literal_eval(row[5])),
756 'weight': row[7], 756 'weight': row[7],
757 'precisions': array(literal_eval(row[8]))}) 757 'precisions': array(literal_eval(row[8]))})
758 if len(gmm) > 0: 758 if len(gmm) > 0:
759 tmp = mixture.GaussianMixture(len(gmm), covarianceType) 759 tmp = mixture.GaussianMixture(n_components=len(gmm), covariance_type=covarianceType)
760 tmp.means_ = array([gaussian['mean'] for gaussian in gmm]) 760 tmp.means_ = array([gaussian['mean'] for gaussian in gmm])
761 tmp.covariances_ = array([gaussian['covar'] for gaussian in gmm]) 761 tmp.covariances_ = array([gaussian['covar'] for gaussian in gmm])
762 tmp.weights_ = array([gaussian['weight'] for gaussian in gmm]) 762 tmp.weights_ = array([gaussian['weight'] for gaussian in gmm])
763 tmp.gmmTypes = [gaussian['type'] for gaussian in gmm] 763 tmp.gmmTypes = [gaussian['type'] for gaussian in gmm]
764 tmp.precisions_cholesky_ = array([gaussian['precisions'] for gaussian in gmm]) 764 tmp.precisions_cholesky_ = array([gaussian['precisions'] for gaussian in gmm])