Python Framework.add_data示例

import os, sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from src.classifier import Framework

c = Framework()
comment = "Multiclassification: 4 classes, V2, +2017 data, jet variable initialization: -5"
# change this line for each run
c.add_comment(comment)
print comment
treePath = 'dimuons/tree'

# path = '/tmp/Hmumu_ntuples/updated/'
path = "/mnt/hadoop/store/user/dkondrat/2017_ntuples/"
# path = '/scratch/browngpu/dkondra/2016_ntuples_updated/'

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

示例#2

显示文件

metadataPath = 'dimuons/metadata'

c.set_tree_path(treePath)
c.set_metadata_path(metadataPath)

c.set_year("2017")

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.name, ZJets_aMC_2017.path,

示例#3

显示文件

文件： test_multi_new.py 项目： kondratyevd/Hmumu_ML

c = Framework()
# comment = "Multiclassification: 4 classes, V2, +2016 data, jet variable initialization: -5"	
				# change this line for each run
# c.add_comment(comment)
# print comment

treePath = 'dimuons/tree'
metadataPath = 'dimuons/metadata'

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

# c.add_data('SingleMu_2016B', path+'SingleMuon/SingleMu_2017B/190327_172730/0000/',   		4823				) # lumi [/pb]
c.add_data('SingleMu_2016B', path+'/SingleMu2017B/',   		4823				) # lumi [/pb]
# 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')