Mercurial Hosting > traffic-intelligence

diff scripts/learn-poi.py @ 927:c030f735c594

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added assignment of trajectories to prototypes and cleanup of insert queries
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 11 Jul 2017 17:56:23 -0400
parents 7345f0d51faa
children 933670761a57
line wrap: on
line diff
--- a/scripts/learn-poi.py	Mon Jul 10 18:04:41 2017 -0400
+++ b/scripts/learn-poi.py	Tue Jul 11 17:56:23 2017 -0400
@@ -65,7 +65,7 @@
         plt.title(gmmType)
         print(gmmType+' Clusters:\n{}'.format(ml.computeClusterSizes(labels, range(models[gmmType].n_components))))
     # save
-    storage.savePOIs(args.databaseFilename, models[gmmType], gmmType, gmmId)
+    storage.savePOIsToSqlite(args.databaseFilename, models[gmmType], gmmType, gmmId)
     # save assignments
     if args.assign:
         for o, l in zip(objects, labels):