Configuration/python/RecoBTag_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # define the b-tag squences for offline reconstruction
0004 from RecoBTag.SoftLepton.softLepton_cff import *
0005 from RecoBTag.ImpactParameter.impactParameter_cff import *
0006 from RecoBTag.SecondaryVertex.secondaryVertex_cff import *
0007 from RecoBTag.Combined.combinedMVA_cff import *
0008 from RecoBTag.CTagging.RecoCTagging_cff import *
0009 from RecoBTag.Combined.deepFlavour_cff import *
0010 from RecoBTag.ONNXRuntime.pfDeepFlavour_cff import *
0011 from RecoBTag.ONNXRuntime.pfDeepDoubleX_cff import *
0012 from RecoBTag.ONNXRuntime.pfDeepBoostedJet_cff import *
0013 from RecoBTag.ONNXRuntime.pfHiggsInteractionNet_cff import *
0014 from RecoBTag.ONNXRuntime.pfParticleNet_cff import *
0015 from RecoBTag.ONNXRuntime.pfParticleNetAK4_cff import *
0016 from RecoBTag.ONNXRuntime.pfParticleTransformerAK4_cff import *
0017 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4_cff import *
0018 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4V1_cff import *
0019 from RecoBTag.ONNXRuntime.pfGlobalParticleTransformerAK8_cff import *
0020 from RecoVertex.AdaptiveVertexFinder.inclusiveVertexing_cff import *
0021 from RecoBTag.PixelCluster.pixelClusterTagInfos_cfi import *
0022
0023 legacyBTaggingTask = cms.Task(
0024     # impact parameters and IP-only algorithms
0025     impactParameterTagInfos,
0026     trackCountingHighEffBJetTags,
0027     jetProbabilityBJetTags,
0028     jetBProbabilityBJetTags,
0029
0030     # SV tag infos depending on IP tag infos, and SV (+IP) based algos
0031     secondaryVertexTagInfos,
0032     simpleSecondaryVertexHighEffBJetTags,
0033     combinedSecondaryVertexV2BJetTags,
0034     inclusiveSecondaryVertexFinderTagInfos,
0035     combinedInclusiveSecondaryVertexV2BJetTags,
0036     ghostTrackVertexTagInfos,
0037     ghostTrackBJetTags,
0038
0039     # soft lepton tag infos and algos
0040     softPFMuonsTagInfos,
0041     softPFMuonBJetTags,
0042     softPFElectronsTagInfos,
0043     softPFElectronBJetTags,
0044
0045     # overall combined taggers
0046     combinedMVAV2BJetTags,
0047
0048     # pixel cluster
0049     pixelClusterTagInfos,
0050 )
0051 legacyBTagging = cms.Sequence(legacyBTaggingTask)
0052
0053 # new candidate-based fwk, with PF inputs
0054 pfBTaggingTask = cms.Task(
0055     # impact parameters and IP-only algorithms
0056     pfImpactParameterTagInfos,
0057     pfTrackCountingHighEffBJetTags,
0058     pfJetProbabilityBJetTags,
0059     pfJetBProbabilityBJetTags,
0060
0061     # SV tag infos depending on IP tag infos, and SV (+IP) based algos
0062     pfSecondaryVertexTagInfos,
0063     pfSimpleSecondaryVertexHighEffBJetTags,
0064     pfCombinedSecondaryVertexV2BJetTags,
0065     inclusiveCandidateVertexingTask,
0066     pfInclusiveSecondaryVertexFinderTagInfos,
0067     pfSimpleInclusiveSecondaryVertexHighEffBJetTags,
0068     pfCombinedInclusiveSecondaryVertexV2BJetTags,
0069     pfGhostTrackVertexTagInfos,
0070     pfGhostTrackBJetTags,
0071     pfDeepCSVTask,
0072     pfDeepFlavourTask,
0073
0074     # soft lepton tag infos and algos
0075     softPFMuonsTagInfos,
0076     softPFMuonBJetTags,
0077     softPFElectronsTagInfos,
0078     softPFElectronBJetTags,
0079
0080     # overall combined taggers
0081     #CSV + soft-lepton + jet probability discriminators combined
0082     pfCombinedMVAV2BJetTags,
0083     pfChargeBJetTags,
0084
0085     # pixel cluster
0086     pixelClusterTagInfos,
0087
0088 )
0089
0090 pfBTagging = cms.Sequence(pfBTaggingTask)
0091
0092 btaggingTask = cms.Task(
0093     pfBTaggingTask,
0094     pfCTaggingTask
0095 )
0096 btagging = cms.Sequence(btaggingTask)
0097
0098 ## modifying b-tagging task in Run3 adding ParticleNet inferece
0099 from Configuration.Eras.Modifier_run3_common_cff import run3_common
0100 _pfBTaggingTask_run3 = cms.Task(
0101     # Keep all the infos and DeepCSV and DeepFlavour
0102     pfImpactParameterTagInfos,
0103     pfTrackCountingHighEffBJetTags,
0104     pfJetProbabilityBJetTags,
0105     pfJetBProbabilityBJetTags,
0106
0107     pfSecondaryVertexTagInfos,
0108     inclusiveCandidateVertexingTask,
0109     pfInclusiveSecondaryVertexFinderTagInfos,
0110     pfGhostTrackVertexTagInfos,
0111     pfDeepCSVTask,
0112     pfDeepFlavourTask,
0113
0114     softPFMuonsTagInfos,
0115     softPFElectronsTagInfos,
0116     pixelClusterTagInfos,
0117
0118     pfParticleNetAK4TaskForRECO,
0119     pfParticleNetTask
0120 )
0121 _pfCTaggingTask_run3 = cms.Task(
0122     inclusiveCandidateVertexingCvsLTask,
0123     pfInclusiveSecondaryVertexFinderCvsLTagInfos,
0124 )
0125 run3_common.toReplaceWith( pfBTaggingTask, _pfBTaggingTask_run3 )
0126 run3_common.toReplaceWith( pfCTaggingTask, _pfCTaggingTask_run3 )
0127