Python Framework.add_dir_to_category Exemples

Exemple #1

Fichier : test_multi.py Projet : kondratyevd/Hmumu_ML

treePath = 'dimuons/tree'

# path = '/tmp/Hmumu_ntuples/updated/'
path = '/scratch/browngpu/dkondra/2016_ntuples_updated/'

c.add_data('SingleMu_2016B'		, path+'/SingleMu_2016B/'	, 5788)
c.add_data('SingleMu_2016C'		, path+'/SingleMu_2016C/'	, 2573)
c.add_data('SingleMu_2016D'		, path+'/SingleMu_2016D/'	, 4248)
c.add_data('SingleMu_2016E'		, path+'/SingleMu_2016E/'	, 4009)
c.add_data('SingleMu_2016F_1'	, path+'/SingleMu_2016F_1/'	, 3102)
c.add_data('SingleMu_2016G'		, path+'/SingleMu_2016G/'	, 7540)
c.add_data('SingleMu_2016H_1'	, path+'/SingleMu_2016H_1/'	, 8392)
c.add_data('SingleMu_2016H_2'	, path+'/SingleMu_2016H_2/'	, 214)

c.add_category('H2Mu_gg', True)
c.add_dir_to_category('H2Mu_gg', path+'/H2Mu_ggH/', 0.009618, 'H2Mu_gg')

c.add_category('H2Mu_VBF', True)
c.add_dir_to_category('H2Mu_VBF', path+'/H2Mu_VBF/', 0.0008208, 'H2Mu_VBF')

c.add_category('ZJets_MG', False)
c.add_dir_to_category('ZJets_MG', path+'/ZJets_MG/', 5765.4, 'ZJets_MG')

c.add_category('tt_ll_AMC', False)
c.add_dir_to_category('tt_ll_AMC', path+'/tt_ll_AMC/', 85.656*0.9, 'tt_ll_AMC')

c.set_tree_path(treePath)

c.add_variable("muPairs.pt", 				1) 
c.add_variable("muPairs.eta", 				1)
c.add_variable("muPairs.dEta", 				1) 

Exemple #2

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.add_dir_to_category(ggH_2017_powheg.name, ggH_2017_powheg.path,
                      ggH_2017_powheg.xSec, c.ggh_label)

c.add_category(c.vbf_label, True)
c.add_dir_to_category(VBF_2017_powheg.name, VBF_2017_powheg.path,
                      VBF_2017_powheg.xSec, c.vbf_label)

c.add_category(c.dy_label, False)
c.add_dir_to_category(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path,
                      ZJets_aMC_2017_hiStat.xSec, c.dy_label)

c.add_category(c.tt_label, False)
c.add_dir_to_category(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
                      tt_ll_POW_2017.xSec, c.tt_label)

##########################################################

Exemple #3

c.set_year("2017")

c.multiclass = False
c.sig_label = "sig"
c.bkg_label = "bkg"

##################### 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.sig_label, True)
c.add_dir_to_category(ggH_2017_powheg.name, ggH_2017_powheg.path,
                      ggH_2017_powheg.xSec, c.sig_label)
c.add_dir_to_category(VBF_2017_powheg.name, VBF_2017_powheg.path,
                      VBF_2017_powheg.xSec, c.sig_label)

c.add_category(c.bkg_label, False)
c.add_dir_to_category(ZJets_aMC_2017.name, ZJets_aMC_2017.path,
                      ZJets_aMC_2017.xSec, c.bkg_label)
c.add_dir_to_category(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
                      tt_ll_POW_2017.xSec, c.bkg_label)

##########################################################

###  ------   Raffaele's variables   ------ ###
c.add_variable("muPairs.pt", 1)
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.dEta", 1)

Exemple #4

Fichier : test_hadoop_2017.py Projet : kondratyevd/Hmumu_ML

c.add_data('SingleMu_2017B',
           path + 'SingleMuon/SingleMu_2017B/180802_163835/0000/', 4823)
c.add_data('SingleMu_2017C',
           path + 'SingleMuon/SingleMu_2017C/180802_163916/0000/', 9664)
c.add_data('SingleMu_2017D',
           path + 'SingleMuon/SingleMu_2017D/180802_163956/0000/', 4252)
c.add_data('SingleMu_2017E',
           path + 'SingleMuon/SingleMu_2017E/180802_164036/0000/', 9278)
c.add_data('SingleMu_2017F',
           path + 'SingleMuon/SingleMu_2017F/180802_164117/0000/',
           (13540 - 916))

c.add_category('H2Mu_gg', True)
c.add_dir_to_category(
    'H2Mu_gg', path +
    '/GluGlu_HToMuMu_M125_13TeV_amcatnloFXFX_pythia8/180802_164158/0000/',
    0.009618, 'H2Mu_gg')

c.add_category('H2Mu_VBF', True)
c.add_dir_to_category(
    'H2Mu_VBF',
    path + '/VBFH_HToMuMu_M125_13TeV_amcatnloFXFX_pythia8/180802_164241/0000/',
    0.0008208, 'H2Mu_VBF')

c.add_category('ZJets_aMC', False)
c.add_dir_to_category(
    'ZJets_aMC', path +
    '/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/180802_165055/0000/',
    5765.4, 'ZJets_aMC')

c.add_category('tt_ll_AMC', False)

Exemple #5

c.set_metadata_path(metadataPath)

c.set_year("inclusive_ucsd")
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_category(c.ggh_label, True)
c.add_dir_to_category(ggH_ucsd_2016.name, ggH_ucsd_2016.path, ggH_ucsd_2016.xSec, c.ggh_label)
c.add_dir_to_category(ggH_ucsd_2017.name, ggH_ucsd_2017.path, ggH_ucsd_2017.xSec, c.ggh_label)
c.add_dir_to_category(ggH_ucsd_2018.name, ggH_ucsd_2018.path, ggH_ucsd_2018.xSec, c.ggh_label)

c.add_category(c.vbf_label, True)
c.add_dir_to_category(VBF_ucsd_2016.name, VBF_ucsd_2016.path, VBF_ucsd_2016.xSec, c.vbf_label)
c.add_dir_to_category(VBF_ucsd_2017.name, VBF_ucsd_2017.path, VBF_ucsd_2017.xSec, c.vbf_label)
c.add_dir_to_category(VBF_ucsd_2018.name, VBF_ucsd_2018.path, VBF_ucsd_2018.xSec, c.vbf_label)

c.add_category(c.dy_label, False)
c.add_dir_to_category(DY_ucsd_2016.name, DY_ucsd_2016.path, DY_ucsd_2016.xSec, c.dy_label)
c.add_dir_to_category(DY_ucsd_2017.name, DY_ucsd_2017.path, DY_ucsd_2017.xSec, c.dy_label)
c.add_dir_to_category(DY_ucsd_2018.name, DY_ucsd_2018.path, DY_ucsd_2018.xSec, c.dy_label)

c.add_category(c.tt_label, False)
c.add_dir_to_category(tt_ucsd_2016.name, tt_ucsd_2016.path, tt_ucsd_2016.xSec, c.tt_label)

Exemple #6

# path = '/scratch/browngpu/dkondra/2016_ntuples_updated/'
path = '/scratch/gilbreth/dkondra/'
# path = "/mnt/hadoop/store/user/dkondrat/"

# c.add_data('SingleMu_2016B'		, path+'/SingleMu_2016B/'	, 5788)
# c.add_data('SingleMu_2016C'		, path+'/SingleMu_2016C/'	, 2573)
# c.add_data('SingleMu_2016D'		, path+'/SingleMu_2016D/'	, 4248)
# c.add_data('SingleMu_2016E'		, path+'/SingleMu_2016E/'	, 4009)
# c.add_data('SingleMu_2016F_1'	, path+'/SingleMu_2016F_1/'	, 3102)
# c.add_data('SingleMu_2016G'		, path+'/SingleMu_2016G/'	, 7540)
# c.add_data('SingleMu_2016H_1'	, path+'/SingleMu_2016H_1/'	, 8392)
# c.add_data('SingleMu_2016H_2'	, path+'/SingleMu_2016H_2/'	, 214)

c.add_category('H2Mu_gg', True)
c.add_dir_to_category(
    'H2Mu_gg', path +
    '/GluGluHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_gg_powheg/190323_193526/0000/',
    0.009618, 'H2Mu_gg')

c.add_category('H2Mu_VBF', True)
c.add_dir_to_category(
    'H2Mu_VBF', path +
    '/VBFHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_VBF_powheg/190322_195711/0000/',
    0.0008208, 'H2Mu_VBF')

c.add_category('ZJets_aMC', False)
c.add_dir_to_category(
    'ZJets_aMC', path +
    '/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/190323_195010/0000/',
    5765.4, 'ZJets_aMC')

c.add_category('tt_ll_POW', False)

Exemple #7

Fichier : test_multi_new.py Projet : kondratyevd/Hmumu_ML

# c.add_data('SingleMu_2016H_2'	, path+'/SingleMu_2016H_2/'	, 214)

# c.add_category('H2Mu_gg', True)
# c.add_dir_to_category('H2Mu_gg', path+'/GluGluHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_gg_powheg/190323_193526/0000/', 0.009618, 'H2Mu_gg')

# c.add_category('H2Mu_VBF', True)
# c.add_dir_to_category('H2Mu_VBF', path+'/VBFHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_VBF_powheg/190322_195711/0000/', 0.0008208, 'H2Mu_VBF')

# c.add_category('ZJets_aMC', False)
# c.add_dir_to_category('ZJets_aMC', path+'/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/190323_195010/0000/', 5765.4, 'ZJets_aMC')

# c.add_category('tt_ll_POW', False)
# c.add_dir_to_category('tt_ll_POW', path+'/TTTo2L2Nu_TuneCP5_PSweights_13TeV-powheg-pythia8/tt_ll_POW/190322_195819/0000/', 85.656, 'tt_ll_POW')

c.add_category('H2Mu_gg', True)
c.add_dir_to_category('H2Mu_gg', path+'/H2Mu_ggH/', 0.009618, 'H2Mu_gg')

c.add_category('H2Mu_VBF', True)
c.add_dir_to_category('H2Mu_VBF', path+'/H2Mu_VBF/', 0.0008208, 'H2Mu_VBF')

c.add_category('ZJets_aMC', False)
c.add_dir_to_category('ZJets_aMC', path+'/ZJets_AMC/', 5765.4, 'ZJets_aMC')

c.add_category('tt_ll_POW', False)
c.add_dir_to_category('tt_ll_POW', path+'/tt_ll_POW/', 85.656, 'tt_ll_POW')

# c.add_signal_dir('H2Mu_gg', path+'/GluGluHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_gg_powheg/190323_193526/0000/', 0.009618)
# c.add_signal_dir('H2Mu_VBF', path+'/VBFHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_VBF_powheg/190322_195711/0000/', 0.0008208)

# c.add_background_dir('ZJets_aMC', path+'/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/190323_195010/0000/', 5765.4)
# c.add_background_dir('tt_ll_POW', path+'/TTTo2L2Nu_TuneCP5_PSweights_13TeV-powheg-pythia8/tt_ll_POW/190322_195819/0000/', 85.656)