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')
