예제 #1
0
split_data_set = False
#conditions_for_splitting = definitions.ttbar_processes()


#----------------------------------------------------------------------------------------------------


save_path = os.path.join(workdir_base, 'HDF5_files')

list_of_generator_level_variables       = excluded_variables.generator_level_variables()
list_of_other_always_excluded_variables = excluded_variables.other_always_excluded_variables()
list_of_vector_variables_lepton         = vector_variables.lepton_variables()
list_of_vector_variables_jet            = vector_variables.jet_variables()


weights_to_keep = definitions.default_weight_list()


number_of_saved_jets    = 6
number_of_saved_leptons = 1


percentage_validation = 20


#----------------------------------------------------------------------------------------------------
if not os.path.isdir(save_path):
    if os.path.isdir(os.path.dirname(save_path)):
        os.mkdir(save_path)
#----------------------------------------------------------------------------------------------------
function_call_dict = {'filename_outputfile'                     : filename_outputfile,
drop_events_negative_weights = True


#----------------------------------------------------------------------------------------------------


save_path = os.path.join(workdir_base, name_subdir, 'data_sets')


path_to_merged_data_set = os.path.join(workdir_base, 'HDF5_files/data_set_merged.hdf')


### Specify the MC weights that will be applied in the training.
### The weights you want to use here have to be included in the "weights_to_keep" list in preprocessing_1_root_to_HDF5.py
### The weights will be normalized so that the sum of weights is equal for each process.
weights_to_be_applied = definitions.default_weight_list()


#----------------------------------------------------------------------------------------------------
if not os.path.isdir(save_path):
    if os.path.isdir(os.path.dirname(save_path)):
        os.mkdir(save_path)
#----------------------------------------------------------------------------------------------------
function_call_dict = {'jet_btag_category'            : jet_btag_category,
                      'selected_processes'           : selected_processes,
                      'binary_classification'        : binary_classification,
                      'select_variables'             : select_variables,
                      'cutbased_event_selection'     : cutbased_event_selection,
                      'create_new_process_labels'    : create_new_process_labels,
                      'save_path'                    : save_path,
                      'path_to_merged_data_set'      : path_to_merged_data_set,