def run_test_ntuple(reduced_dataset, weights_file, output_file, output_tree = 'performance', print_and_exit = False): output_dir = os.path.dirname(output_file) if not os.path.isdir(output_dir): os.makedirs(output_dir) pt_categories, eta_categories = _get_pt_eta_categories(reduced_dataset) if print_and_exit: pt_categories = [1.0,20.3] eta_categories = [0.0,1.0,3.5] pynn.makeNtuple(weights_file = weights_file, reduced_dataset = reduced_dataset, output_file = output_file, output_tree = output_tree, debug = print_and_exit)
# Author: Daniel Guest ([email protected]) import sys, dl, os, glob # without this root has trouble # binding_rules = dl.RTLD_GLOBAL | dl.RTLD_NOW # sys.setdlopenflags(binding_rules) import pynn if len(sys.argv) > 1: reduced_dataset = sys.argv[1] else: sys.exit('give a ds to run on') output_directory = 'test' # pynn.trainNN(reduced_dataset = reduced_dataset, # output_directory = output_directory, # n_iterations = 10, # dilution_factor = 2, # use_sd = False, # with_ip3d = True, # nodes = (10,10), # debug = False) pynn.makeNtuple(weights_file = output_directory + '/weightMinimum.root', reduced_dataset = reduced_dataset, output_file = output_directory + '/test_ntuple.root')