Exemplo n.º 1
0
elif options.tauCluster == 7:
    print " ====== use slimmedTausElectronCleanedTight cluster ======"
    from MuMuTauTauTreeMaker.MuTauTreelizer.TauIdMVA_slimmedTausElectronCleanedTight import *
    myTool = TauIDEmbedder(process, cms, debug=True, toKeep=["2017v2"])
    myTool.runTauID()
    process.rerunTauIDSequence = cms.Sequence(
        process.rerunMvaIsolationSequence * process.slimmedTausNewID)

elif options.tauCluster == 8:
    print " ====== use slimmedTaus DeepID cluster ======"
    updatedTauName = "slimmedTausNewID"
    import MuMuTauTauTreeMaker.MuTauTreelizer.TauIdDeep_slimmedTaus as tauIdConfig
    tauIdEmbedder = tauIdConfig.TauIDEmbedder(process,
                                              cms,
                                              debug=False,
                                              updatedTauName=updatedTauName,
                                              toKeep=["deepTau2017v2p1"])
    tauIdEmbedder.runTauID()
    process.rerunTauIDSequence = cms.Sequence(
        process.rerunMvaIsolationSequence * getattr(process, updatedTauName))

elif options.tauCluster == 9:
    print " ====== use slimmedTausMuonCleaned DeepID cluster ======"
    updatedTauName = "slimmedTausNewID"
    import MuMuTauTauTreeMaker.MuTauTreelizer.TauIdDeep_slimmedTausMuonCleaned as tauIdConfig
    tauIdEmbedder = tauIdConfig.TauIDEmbedder(process,
                                              cms,
                                              debug=False,
                                              updatedTauName=updatedTauName,
                                              toKeep=["deepTau2017v2p1"])
Exemplo n.º 2
0
process.MessageLogger.cerr.FwkReport.reportEvery = cms.untracked.int32(1000)

# --- please specify the sample that you need to run for local test ---
process.source = cms.Source("PoolSource",
                            fileNames=cms.untracked.vstring(
                                options.inputFiles))

######### embed 2017v2 tauID into the miniAOD ###############
# reference: https://twiki.cern.ch/twiki/bin/view/CMSPublic/SWGuidePFTauID#Rerunning_of_the_tau_ID_on_M_AN1
if options.tauCluster <= 0:
    print " ====== use slimmedTaus cluster ======"
    updatedTauName = "slimmedTausNewID"
    import MuMuTauTauTreeMaker.MuTauTreelizer.TauIdDeep_slimmedTaus as tauIdConfig
    tauIdEmbedder = tauIdConfig.TauIDEmbedder(
        process,
        cms,
        debug=True,
        updatedTauName=updatedTauName,
        toKeep=["deepTau2017v2p1", "2017v2"])
    tauIdEmbedder.runTauID()
    process.rerunTauIDSequence = cms.Sequence(
        process.rerunMvaIsolationSequence * getattr(process, updatedTauName))

elif options.tauCluster == 1:
    print " ====== use slimmedTausMuonCleaned cluster ======"
    updatedTauName = "slimmedTausNewID"
    import MuMuTauTauTreeMaker.MuTauTreelizer.TauIdDeep_slimmedTausMuonCleaned as tauIdConfig
    tauIdEmbedder = tauIdConfig.TauIDEmbedder(
        process,
        cms,
        debug=True,
        updatedTauName=updatedTauName,