def traj_conv(): global traj_conv_ if not traj_conv_: traj_conv_ = createTrajectoryConversion(graph_type=graph_type, process=experiment_design['trajectory_conversion']['process'], params=traj_conv_param, network=net, max_nb_mixture=traj_conv_param['max_n_modes']) return traj_conv_
def fillTrajectoryCache(graph_type,basic_geometry,data_source,traj_conv_description,n_jobs=1): net = get_network(**basic_geometry) tic("Loaded network = {0} links".format(len(net)), "fillTrajectoryCache") traj_conv = createTrajectoryConversion(graph_type=graph_type, process=traj_conv_description['process'], params=traj_conv_description['params'], network=net, max_nb_mixture=traj_conv_description['params']['max_n_modes'], n_jobs=n_jobs) dates = data_source['dates'] from joblib import Parallel, delayed Parallel(n_jobs=n_jobs)(delayed(wrapper)(data_source['feed'], basic_geometry['nid'], date, basic_geometry['net_type'], basic_geometry['box'], traj_conv_description, traj_conv, net) for date in dates)
""" Starting the main procedure. TODO: put all learning in a function """ # pylint:disable=W0142 experiment_design = experiment_design experiment_name = experiment_design['name'] # Get the network basic_geometry = experiment_design['basic_geometry'] net = get_network(**basic_geometry) graph_type = experiment_design['graph_type'] traj_conv_param = experiment_design['trajectory_conversion']['params'] traj_conv = createTrajectoryConversion(graph_type=graph_type, process=experiment_design['trajectory_conversion']['process'], params=traj_conv_param, network=net, max_nb_mixture=traj_conv_param['max_n_modes']) traj_conv_one_mode = createTrajectoryConversion(graph_type=graph_type, process='mixture_auto', params=traj_conv_param, network=net, max_nb_mixture=1) # mode_counts = dict([(link_id,1) for link_id in net.keys()]) if graph_type == 'simple': hmm_graph = model.createHMMGraphFromNetwork(net, mode_counts=traj_conv.modeCounts()) hmm_graph_one_mode = model.createHMMGraphFromNetwork(net, mode_counts=traj_conv_one_mode.modeCounts()) # hmm_graph = model.createHMMGraphFromNetwork(net, mode_counts=mode_counts) else: