Пример #1
0
def run_performance(reduced_dataset, weights_file, 
            output_file = 'hists/performance_plots.root', 
            with_ip3d = True, 
            print_and_exit = False): 


    output_dir = os.path.dirname(output_file)
    if not os.path.isdir(output_dir): 
        os.makedirs(output_dir)

    pynn.testNN(reduced_dataset = reduced_dataset, 
                weights_file = weights_file, 
                # dilution_factor = 2, 
                use_sd = False, 
                with_ip3d = with_ip3d, 
                debug = print_and_exit, 
                output_file = output_file) 
Пример #2
0
    input_ds = sys.argv[1]

with_ip3d = True
# if len(sys.argv) > 2: 
#     if 'True' in sys.argv[2:]: 
#         with_ip3d = True

trained_nn = 'ip3d_weights/weightMinimum.root'
if not os.path.isfile(trained_nn): 
    sys.exit("ERROR: no file %s found" % trained_nn)

out_path = 'draw/all_plots.root'
out_dir, out_name  = os.path.split(out_path)
if out_dir and not os.path.isdir(out_dir): 
    os.mkdir(out_dir)

full_ds_name = 'reduceddataset_%s_forNN.root' % input_ds
full_path = '../reduceddatasets/' + full_ds_name

if not os.path.isfile(full_path): 
    sys.exit('no file to read!')

pynn.testNN(input_file = full_path, 
            trained_nn_file = trained_nn, 
            # dilution_factor = 2, 
            use_sd = False, 
            with_ip3d = with_ip3d, 
            debug = False, 
            out_file = out_path)