ttbar_categories = root2pandas.EventCategories()
ttbar_categories.addCategory(
"ttbb", selection="(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"tt2b", selection="(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"ttb", selection="(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"ttlf", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"ttcc", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 1)")
# initialize dataset class
dataset = root2pandas.Dataset(
outputdir="/nfs/dust/cms/user/vdlinden/DNNInputFiles/DNN_ttH_2017/",
naming="_dnn_newJEC",
addCNNmap=False,
addMEM=True)
# add base event selection
dataset.addBaseSelection(base_selection)
ntuplesPath = "/nfs/dust/cms/user/kelmorab/ttH_2018/ntuples_forDNN_v4/"
memPath = "/nfs/dust/cms/user/vdlinden/MEM_2017/"
# add samples to dataset
dataset.addSample(
sampleName="ttHbb",
ntuples=ntuplesPath +
"/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
categories=ttH_categories,
ttbb_categories = root2pandas.EventCategories()
ttbb_categories.addCategory("ttbb", selection = "(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0)")
ttbb_categories.addCategory("tt2b", selection = "(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)")
ttbb_categories.addCategory("ttb", selection = "(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)")
ttbb_categories.addCategory("ttb_bb", selection = "(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0) or \
(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)")
ttbb_categories.addCategory("tthf", selection = "(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0) or \
(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0) or \
(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)")
# initialize dataset class
dataset = root2pandas.Dataset(
outputdir = outputdir,
naming = options.Name,
addMEM = False,#options.MEM,
maxEntries = options.maxEntries)
# add base event selection
dataset.addBaseSelection(base_selection)
ntuplesPath = "/nfs/dust/cms/user/swieland/ttH_legacy/ntuple/2017/"
memPath = "/nfs/dust/cms/user/mwassmer/ttH_2018/MEMs_v2/" #outdated
# add samples to dataset
dataset.addSample(
sampleName = "ttHbb",
ntuples = ntuplesPath+"/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
ttbar_categories = root2pandas.EventCategories()
ttbar_categories.addCategory(
"ttbb", selection="(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"tt2b", selection="(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"ttb", selection="(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"ttlf", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 0)")
ttbar_categories.addCategory(
"ttcc", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 1)")
# initialize dataset class
dataset = root2pandas.Dataset(
outputdir="/nfs/dust/cms/user/vdlinden/DNNInputFiles/ttZ_DNN_v2/",
naming="dnn",
addCNNmap=False,
addMEM=False)
# add base event selection
dataset.addBaseSelection(base_selection)
ntuplesPath = "/nfs/dust/cms/user/vdlinden/ttH_2018/ntuples/ntuples_v5_forDNN/"
ttZntuples = "/nfs/dust/cms/user/vdlinden/ttZ_2019/ntuples_v1/"
memPath = "/nfs/dust/cms/user/vdlinden/MEM_2017/"
# add samples to dataset
dataset.addSample(
sampleName="ttHbb",
ntuples=ntuplesPath +
"/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
if not os.path.isabs(options.outputDir):
outputdir = basedir + "/workdir/" + options.outputDir
elif os.path.exists(options.outputDir) or os.path.exists(
os.path.dirname(options.outputDir)):
outputdir = options.outputDir
else:
sys.exit("ERROR: Output Directory does not exist!")
# define a base event selection which is applied for all Samples
# initialize dataset class
dataset = root2pandas.Dataset(
outputdir=outputdir,
naming=options.Name,
addMEM=options.MEM,
maxEntries=options.maxEntries,
tree=options.treeName,
varName_Run="runNumber",
varName_LumiBlock="lumiBlock",
varName_Event="eventNumber",
)
ntuplesPath = "/nfs/dust/cms/user/missirol/sandbox/ttHbb/output_190914_DeepJet/2016/exe1/selectionRoot_reco_liteTreeTTH/Nominal"
ttH_categories = root2pandas.EventCategories()
ttH_categories.addCategory("ttH", selection=None)
ttbar_categories = root2pandas.EventCategories()
ttbar_categories.addCategory("ttbar", selection=None)
ttbar_bb_categories = root2pandas.EventCategories()
ttbar_bb_categories.addCategory("ttbb", selection=None)
base_selection = "(" + base + " and (" + single_mu_sel + " or " + single_el_sel + "))"
# define output classes
ttH_categories = root2pandas.EventCategories()
ttH_categories.addCategory("ttH", selection=None)
ntuplespath = "/nfs/dust/cms/user/vdlinden/legacyTTH/ntuples/ntuples_ttZ/"
friendTrees = {
"RecoDNN_Z":
"/nfs/dust/cms/user/vdlinden/legacyTTH/karim/workdir/ntuples/RecoDNN_Z"
}
# initialize dataset class
dataset = root2pandas.Dataset(outputdir=outputdir,
naming=options.Name,
addMEM=options.MEM,
maxEntries=options.maxEntries,
ttbarReco=options.ttbarReco,
friendTrees=friendTrees,
ncores=options.ncores)
# add base event selection
dataset.addBaseSelection(base_selection)
# add samples to dataset
dataset.addSample(
sampleName="ttH",
ntuples=ntuplespath +
"/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
categories=ttH_categories,
lumiWeight=41.5,
)
# define a base event selection which is applied for all Samples
base_selection = "(\
(N_LooseMuons == 0 and N_TightElectrons == 1) \
or \
(N_LooseElectrons == 0 and N_TightMuons == 1) \
)"
categories = root2pandas.EventCategories()
categories.addCategory("ttbar", selection = None)
# initialize dataset class
dataset = root2pandas.Dataset(
outputdir = "/nfs/dust/cms/user/vdlinden/DNNInputFiles/ttbarMatcher/",
naming = "input",
addCNNmap = False,
addMEM = False,
maxEntries = 200000)
# add base event selection
dataset.addBaseSelection(base_selection)
# add samples to dataset
dataset.addSample(
sampleName = "ttbar",
ntuples = "/nfs/dust/cms/user/vdlinden/ttH_2018/ntuples/ntuples_ttbarMatching/TTToSemiLeptonic_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
categories = categories)
# initialize variable list
dataset.addAllVariablesNoIndex()
