Mercurial Hosting > traffic-intelligence

comparison scripts/learn-motion-patterns.py @ 848:0cb69238e6f5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected load/save prototypes
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 21 Jul 2016 17:51:11 -0400
parents 5a68779d7777
children a414a7d58483
comparison
equal deleted inserted replaced
847:36c5bee9a887 848:0cb69238e6f5
51 prototypeIndices, labels = ml.prototypeCluster(trajectories, similarities, args.minSimilarity, lambda x,y : lcss.computeNormalized(x, y), args.minClusterSize, args.randomInitialization) # this line can be called again without reinitializing similarities 51 prototypeIndices, labels = ml.prototypeCluster(trajectories, similarities, args.minSimilarity, lambda x,y : lcss.computeNormalized(x, y), args.minClusterSize, args.randomInitialization) # this line can be called again without reinitializing similarities
52 52
53 clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1) 53 clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1)
54 print(clusterSizes) 54 print(clusterSizes)
55 55
56 storage.savePrototypesToSqlite(args.databaseFilename, [objects[i].getNum() for i in prototypeIndices], args.trajectoryType, {objects[i].getNum():clusterSizes[i] for i in prototypeIndices}) 56 storage.savePrototypesToSqlite(args.databaseFilename, [objects[i].getNum() for i in prototypeIndices], args.trajectoryType, {objects[i].getNum():clusterSizes[i] for i in prototypeIndices}) # if saving filenames, add for example {objects[i].getNum():objects[i].dbFilename for i in prototypeIndices}
57 57
58 if args.saveSimilarities: 58 if args.saveSimilarities:
59 np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f') 59 np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f')
60 60
61 if args.display: 61 if args.display: