Python Framework.add_mc_dir_to_evaluate示例

示例#1

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

gInterpreter.ProcessLine('#include "interface/GenParentInfo.h"')
gInterpreter.ProcessLine('#include "interface/GenMuonInfo.h"')
gInterpreter.ProcessLine('#include "interface/GenMuPairInfo.h"')
gInterpreter.ProcessLine('#include "interface/GenJetInfo.h"')

c = Framework()
treePath = 'dimuons/tree'

c.add_data_to_evaluate(SingleMu2017B.name, SingleMu2017B.path)
c.add_data_to_evaluate(SingleMu2017C.name, SingleMu2017C.path)
c.add_data_to_evaluate(SingleMu2017D.name, SingleMu2017D.path)
c.add_data_to_evaluate(SingleMu2017E.name, SingleMu2017E.path)
c.add_data_to_evaluate(SingleMu2017F.name, SingleMu2017F.path)

c.add_mc_dir_to_evaluate(ZJets_aMC_2017_hiStat.name,
                         ZJets_aMC_2017_hiStat.path,
                         ZJets_aMC_2017_hiStat.xSec)
# c.add_mc_dir_to_evaluate(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path, ZJets_aMC_2017_hiStat.xSec)
c.add_mc_dir_to_evaluate(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
                         tt_ll_POW_2017.xSec)
c.add_mc_dir_to_evaluate(ggH_2017_powheg.name, ggH_2017_powheg.path,
                         ggH_2017_powheg.xSec)
c.add_mc_dir_to_evaluate(VBF_2017_powheg.name, VBF_2017_powheg.path,
                         VBF_2017_powheg.xSec)

c.set_tree_path(treePath)

c.add_variable("muPairs.pt", 1)
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.cosThetaCS", 1)
c.add_variable("muPairs.phiCS", 1)

示例#2

gInterpreter.ProcessLine('#include "interface/GenParentInfo.h"')
gInterpreter.ProcessLine('#include "interface/GenMuonInfo.h"')
gInterpreter.ProcessLine('#include "interface/GenMuPairInfo.h"')
gInterpreter.ProcessLine('#include "interface/GenJetInfo.h"')

c = Framework()
treePath = 'dimuons/tree'

# c.add_data_to_evaluate(SingleMu2017B.name, SingleMu2017B.path+"/*root")
# c.add_data_to_evaluate(SingleMu2017C.name, SingleMu2017C.path+"/*root")
# c.add_data_to_evaluate(SingleMu2017D.name, SingleMu2017D.path+"/*root")
# c.add_data_to_evaluate(SingleMu2017E.name, SingleMu2017E.path+"/*root")
# c.add_data_to_evaluate(SingleMu2017F.name, SingleMu2017F.path+"/*root")

# c.add_mc_to_evaluate(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path+"/*root", ZJets_aMC_2017_hiStat.xSec)
c.add_mc_dir_to_evaluate(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path, ZJets_aMC_2017_hiStat.xSec)
# c.add_mc_to_evaluate(tt_ll_POW_2017.name, tt_ll_POW_2017.path+"/*root", tt_ll_POW_2017.xSec)
# c.add_mc_to_evaluate(ggH_2017_powheg.name, ggH_2017_powheg.path+"/*root", ggH_2017_powheg.xSec)
# c.add_mc_to_evaluate(VBF_2017_powheg.name, VBF_2017_powheg.path+"/*root", VBF_2017_powheg.xSec)



c.set_tree_path(treePath)

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

c.add_variable("mu1_pt_Roch_over_mass", 	1)