def load_pace_vectors(dates, consistent_link_set):
# Map (begin_node,connecting_node) --> ID in the pace vector
link_id_map = defaultdict(lambda: -1) # -1 indicates an invalid ID number
for i in xrange(len(consistent_link_set)):
# print i
key = consistent_link_set[i]
# print long(key)
key = long(key)
link_id_map[key] = i
db_main.connect("db_functions/database.conf")
vects = []
weights = []
for date in dates:
# Initialize to zero
# print date
vect = matrix(zeros((len(consistent_link_set), 1)))
weight = matrix(zeros((len(consistent_link_set), 1)))
# Get the travel times for this datetime
curs = db_travel_times.get_travel_times_cursor_new(date)
# Assign travel times into the vector, if this link is in the consistant link set
for (link_id, date_time, paces, num_cars) in curs:
# print (link_id, date_time, paces, num_cars)
# print "link_id",link_id
i = link_id_map[link_id] # i will be -1 if the link is not in the consistant link set
if i >= 0:
vect[i] = paces
weight[i] = num_cars
vects.append(vect)
weights.append(weight)
db_main.close()
return vects, weights
def load_pace_vectors(dates, consistent_link_set):
# Map (begin_node,connecting_node) --> ID in the pace vector
link_id_map = defaultdict(lambda : -1) # -1 indicates an invalid ID number
for i in xrange(len(consistent_link_set)):
# print i
key = consistent_link_set[i]
# print long(key)
key = long(key)
link_id_map[key] = i
db_main.connect('db_functions/database.conf')
vects = []
weights = []
for date in dates:
# Initialize to zero
# print date
vect = matrix(zeros((len(consistent_link_set), 1)))
weight = matrix(zeros((len(consistent_link_set), 1)))
# Get the travel times for this datetime
curs = db_travel_times.get_travel_times_cursor_new(date)
# Assign travel times into the vector, if this link is in the consistant link set
for (link_id, date_time, paces, num_cars) in curs:
# print (link_id, date_time, paces, num_cars)
# print "link_id",link_id
i = link_id_map[link_id] # i will be -1 if the link is not in the consistant link set
if(i>=0):
vect[i] = paces
weight[i] = num_cars
vects.append(vect)
weights.append(weight)
db_main.close()
return vects, weights
def compute_link_counts(dates):
num_appearances = defaultdict(float)
db_main.connect("db_functions/database.conf")
for date in dates:
curs = db_travel_times.get_travel_times_cursor_new(date)
for [link_id, date_time, paces, num_cars] in curs:
num_appearances[link_id] += 1
db_main.close()
return num_appearances
def compute_link_counts(dates):
num_appearances = defaultdict(float)
db_main.connect('db_functions/database.conf')
for date in dates:
curs = db_travel_times.get_travel_times_cursor_new(date)
for [link_id, date_time, paces, num_cars] in curs:
num_appearances[link_id] += 1
db_main.close()
return num_appearances
