Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/learn-poi.py @ 1028:cc5cb04b04b0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
major update using the trafficintelligence package name and install through pip
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 15 Jun 2018 11:19:10 -0400
parents 933670761a57
children
comparison
equal deleted inserted replaced
1027:6129296848d3 1028:cc5cb04b04b0
4 4
5 import numpy as np 5 import numpy as np
6 from sklearn import mixture 6 from sklearn import mixture
7 import matplotlib.pyplot as plt 7 import matplotlib.pyplot as plt
8 8
9 import storage, ml 9 from trafficintelligence import storage, ml
10 10
11 parser = argparse.ArgumentParser(description='The program learns and displays Gaussians fit to beginnings and ends of object trajectories (based on Mohamed Gomaa Mohamed 2015 PhD).') 11 parser = argparse.ArgumentParser(description='The program learns and displays Gaussians fit to beginnings and ends of object trajectories (based on Mohamed Gomaa Mohamed 2015 PhD).')
12 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True) 12 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True)
13 parser.add_argument('-t', dest = 'trajectoryType', help = 'type of trajectories to display', choices = ['feature', 'object'], default = 'object') 13 parser.add_argument('-t', dest = 'trajectoryType', help = 'type of trajectories to display', choices = ['feature', 'object'], default = 'object')
14 parser.add_argument('-n', dest = 'nObjects', help = 'number of objects to display', type = int) 14 parser.add_argument('-n', dest = 'nObjects', help = 'number of objects to display', type = int)