lambda x: abs(x.pdgId) == 13, lambda x: x.TMOneStationTightMuonId, lambda x: x.trackerLayers > 5, lambda x: x.pixelLayers > 0, lambda x: abs(x.dxy) < 0.3 and abs(x.dz) < 20., ]))) #--- Lepton MVA in friend tree from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend MODULES.append(('LepMVAFriendTTH', lambda: LepMVAFriend( ( os.environ["CMSSW_BASE"] + "/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml", os.environ["CMSSW_BASE"] + "/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml", ), training="forMoriond16", label="TTHMoriond16"))) MODULES.append(('LepMVAFriendSUSY', lambda: LepMVAFriend( ( os.environ["CMSSW_BASE"] + "/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml", os.environ["CMSSW_BASE"] + "/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml", ), training="forMoriond16", label="SUSY"))) MODULES.append(('LepMVAFriendJetLessIVF', lambda: LepMVAFriend(( os.environ["CMSSW_BASE"] +
#from CMGTools.TTHAnalysis.tools.finalMVA_susy_2lss import FinalMVA_SUSY_2LSS #MODULES.append( ('finalMVA_susy_2lss', FinalMVA_SUSY_2LSS()) ) #from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco_MC #MODULES.append( ('ttreco_mc', TTEventReco_MC()) ) #from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco #MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"", "BestBySum4NoTJJb":"_bySum4"})) ) #MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"","ByGuessLL2B":"_byLL"})) ) #from CMGTools.TTHAnalysis.tools.MuonMVAFriend import MuonMVAFriend #MODULES.append( ('MuMVAId', MuonMVAFriend("BPH", "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_muMVAId_v1/train70XBPH_BDTG.weights.xml", label="BPH")) ) #MODULES.append( ('MuMVAId', MuonMVAFriend("BPHCalo", "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_muMVAId_v1/train70XBPHCalo_BDTG.weights.xml", label="BPHCalo")) ) #MODULES.append( ('MuMVAId', MuonMVAFriend("Full", "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_muMVAId_v1/train70XFull_BDTG.weights.xml", label="Full")) ) from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend MODULES.append(( 'LepMVAFriend', LepMVAFriend(( "/afs/cern.ch/user/g/gpetrucc/w/TREES_72X_171214/0_lepMVA_v1/%s_BDTG.weights.xml", "/afs/cern.ch/user/g/gpetrucc/w/TREES_72X_171214/0_lepMVA_v1/%s_BDTG.weights.xml" )))) #MODULES.append( ('LepMVAFriend', LepMVAFriend(("/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_lepMVA_v1/SV_%s_BDTG.weights.xml", # "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_lepMVA_v1/SV_%s_BDTG.weights.xml",), # training="muMVAId_SV", label="SV")) ) #from CMGTools.TTHAnalysis.tools.eventVars_MT2 import EventVarsMT2 #MODULES.append( ('MT2', EventVarsMT2()) ) class VariableProducer(Module): def __init__(self, name, booker, modules): Module.__init__(self, name, booker) self._modules = modules
#--- TTH event variables from CMGTools.TTHAnalysis.tools.eventVars_2lss import EventVars2LSS MODULES.append( ('ttH2lss', lambda : EventVars2LSS()) ) from CMGTools.TTHAnalysis.tools.kinMVA_2D_2lss_3l import KinMVA_2D_2lss_3l MODULES.append( ('kinMVA_2D_2lss_3l', lambda : KinMVA_2D_2lss_3l(os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/kinMVA/tth/%s_BDTG.weights.xml")) ) from CMGTools.TTHAnalysis.tools.BDT2_HadTop import BDT2_HadTop MODULES.append( ('BDT2_HadTop', lambda : BDT2_HadTop(os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/kinMVA/tth/TMVAClassification_BDTG.weights_BDT2.xml")) ) #--- Lepton MVA in friend tree from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend MODULES.append( ('LepMVAFriendTTH', lambda: LepMVAFriend((os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml", os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml",), training="forMoriond16", label="TTHMoriond16")) ) MODULES.append( ('LepMVAFriendSUSY', lambda: LepMVAFriend((os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml", os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml",), training="forMoriond16", label="TTZMoriond16")) ) class VariableProducer(Module): def __init__(self,name,booker,modules): Module.__init__(self,name,booker) self._modules = [ (n,m() if type(m) == types.FunctionType else m) for (n,m) in modules ] def beginJob(self): self.t = PyTree(self.book("TTree","t","t")) self.branches = {} for name,mod in self._modules: print name print mod.listBranches()
#MODULES.append( ('3l_mva', FinalMVA_3L()) ) #from CMGTools.TTHAnalysis.tools.bbvars import bbVars #MODULES.append( ('bbvars', bbVars()) ) #from CMGTools.TTHAnalysis.tools.finalMVA_2lss_2 import FinalMVA_2LSS_2 #MODULES.append( ('finalMVA_2lss_2', FinalMVA_2LSS_2()) ) #from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco_MC #MODULES.append( ('ttreco_mc', TTEventReco_MC()) ) #from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco #MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"", "BestBySum4NoTJJb":"_bySum4"})) ) #MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"","ByGuessLL2B":"_byLL"})) ) from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend MODULES.append(( 'LepMVAFriend', LepMVAFriend( "/afs/cern.ch/user/b/botta/CMGToolsGit/newRecipe/CMSSW_5_3_19/src/CMGTools/TTHAnalysis/macros/leptons" ))) class VariableProducer(Module): def __init__(self, name, booker, modules): Module.__init__(self, name, booker) self._modules = modules def beginJob(self): self.t = PyTree(self.book("TTree", "t", "t")) self.branches = {} for name, mod in self._modules: for B in mod.listBranches(): # don't add the same branch twice if B in self.branches: