comparison 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
comparison
equal deleted inserted replaced
1203:7b3384a8e409 1204:a12d126346ff
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])