def main(input_directory, num_epochs=5): print("Loading data...") data = load_datasets(input_directory) print("Building model...") input_var = theano.tensor.tensor4('inputs') target_var = theano.tensor.ivector('targets') network = build_network(input_var) model = build_model(input_var, target_var, network) print("Starting training...") train_network(data, model, num_epochs) print("\nLearned parameters:") params = numpy.array(lasagne.layers.get_all_param_values(network)) param_file = 'output/{}.npy'.format(str(uuid.uuid4())) print("Parameters written to {}".format(param_file)) params.dump(param_file)