c = Framework()
c.label = "bdt_dnn_multi_hiStat"
comment = "Option 1.1: 4-class DNN, mass window [110, 150], w/o even-by-event mass res., ZJets_AMC_hiStat ( mass 105-160 ), w/o CS variables"
# change this line for each run
c.add_comment(comment)
print comment
c.outDir = '/tmp/dkondrat/ML_output/'
treePath = 'dimuons/tree'
metadataPath = 'dimuons/metadata'
c.set_tree_path(treePath)
c.set_metadata_path(metadataPath)
c.set_year("2017")
c.massWindow = [110, 150]
c.multiclass = True
c.dy_label = "ZJets_aMC"
c.tt_label = "tt_ll_POW"
c.ggh_label = "H2Mu_gg"
c.vbf_label = "H2Mu_VBF"
##################### Input samples #######################
c.add_data(SingleMu2017B.name, SingleMu2017B.path, SingleMu2017B.lumi)
c.add_data(SingleMu2017C.name, SingleMu2017C.path, SingleMu2017C.lumi)
c.add_data(SingleMu2017D.name, SingleMu2017D.path, SingleMu2017D.lumi)
c.add_data(SingleMu2017E.name, SingleMu2017E.path, SingleMu2017E.lumi)
c.add_data(SingleMu2017F.name, SingleMu2017F.path, SingleMu2017F.lumi)
c.add_category(c.ggh_label, True)
c = Framework()
c.label = "bdt_dnn_multi_hiStat_m120To130"
comment = "Option 13: 4-class DNN, w/o even-by-event mass res., ZJets_AMC_hiStat ( mass 105-160 ), training mass window: [120, 130]; trying to balance datasets"
# change this line for each run
c.add_comment(comment)
print comment
c.outDir = '/tmp/dkondrat/ML_output/'
treePath = 'dimuons/tree'
metadataPath = 'dimuons/metadata'
c.set_tree_path(treePath)
c.set_metadata_path(metadataPath)
c.set_year("2017")
c.massWindow = [120, 130]
c.multiclass = True
c.dy_label = "ZJets_aMC"
c.tt_label = "tt_ll_POW"
c.ggh_label = "H2Mu_gg"
c.vbf_label = "H2Mu_VBF"
##################### Input samples #######################
c.add_data(SingleMu2017B.name, SingleMu2017B.path, SingleMu2017B.lumi)
c.add_data(SingleMu2017C.name, SingleMu2017C.path, SingleMu2017C.lumi)
c.add_data(SingleMu2017D.name, SingleMu2017D.path, SingleMu2017D.lumi)
c.add_data(SingleMu2017E.name, SingleMu2017E.path, SingleMu2017E.lumi)
c.add_data(SingleMu2017F.name, SingleMu2017F.path, SingleMu2017F.lumi)
c.add_category(c.ggh_label, True)
