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: