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