ONNXRuntime/python/pfParticleTransformerAK4_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoBTag.FeatureTools.pfParticleTransformerAK4TagInfos_cfi import pfParticleTransformerAK4TagInfos
0004
0005 from RecoBTag.ONNXRuntime.pfParticleTransformerAK4JetTags_cfi import pfParticleTransformerAK4JetTags
0006 from RecoBTag.ONNXRuntime.pfParticleTransformerAK4DiscriminatorsJetTags_cfi import pfParticleTransformerAK4DiscriminatorsJetTags
0007 from RecoBTag.ONNXRuntime.pfParticleTransformerAK4SonicJetTags_cfi import pfParticleTransformerAK4SonicJetTags as _pfParticleTransformerAK4SonicJetTags
0008 from CommonTools.PileupAlgos.Puppi_cff import puppi
0009 from CommonTools.RecoAlgos.primaryVertexAssociation_cfi import primaryVertexAssociation
0010 from Configuration.ProcessModifiers.particleTransformerAK4SonicTriton_cff import particleTransformerAK4SonicTriton
0011
0012
0013 particleTransformerAK4SonicTriton.toReplaceWith(pfParticleTransformerAK4JetTags, _pfParticleTransformerAK4SonicJetTags.clone(
0014     Client = cms.PSet(
0015         timeout = cms.untracked.uint32(300),
0016         mode = cms.string("Async"),
0017         modelName = cms.string("particletransformer_AK4"),
0018         modelConfigPath = cms.FileInPath("RecoBTag/Combined/data/models/particletransformer_AK4/config.pbtxt"),
0019         modelVersion = cms.string(""),
0020         verbose = cms.untracked.bool(False),
0021         allowedTries = cms.untracked.uint32(0),
0022         useSharedMemory = cms.untracked.bool(True),
0023         compression = cms.untracked.string(""),
0024     ),
0025     flav_names = pfParticleTransformerAK4JetTags.flav_names,
0026 ))
0027
0028 # declare all the discriminators
0029 # probs
0030 _pfParticleTransformerAK4JetTagsProbs = ['pfParticleTransformerAK4JetTags:' + flav_name
0031                                  for flav_name in pfParticleTransformerAK4JetTags.flav_names]
0032 # meta-taggers
0033 _pfParticleTransformerAK4JetTagsMetaDiscrs = ['pfParticleTransformerAK4DiscriminatorsJetTags:' + disc.name.value()
0034                                       for disc in pfParticleTransformerAK4DiscriminatorsJetTags.discriminators]
0035 _pfParticleTransformerAK4JetTagsAll = _pfParticleTransformerAK4JetTagsProbs + _pfParticleTransformerAK4JetTagsMetaDiscrs
0036
0037
0038
0039 # ==
0040 # This task is not used, useful only if we run it from RECO jets (RECO/AOD)
0041 pfParticleTransformerAK4Task = cms.Task(puppi, primaryVertexAssociation,
0042                              pfParticleTransformerAK4TagInfos, pfParticleTransformerAK4JetTags,
0043                              pfParticleTransformerAK4DiscriminatorsJetTags)
0044 # run from MiniAOD instead
0045 pfParticleTransformerAK4FromMiniAODTask = cms.Task(pfParticleTransformerAK4TagInfos,
0046                              pfParticleTransformerAK4JetTags,
0047                              pfParticleTransformerAK4DiscriminatorsJetTags)
0048
0049 # === Negative tags ===
0050 pfNegativeParticleTransformerAK4TagInfos = pfParticleTransformerAK4TagInfos.clone(
0051     flip = True,
0052     secondary_vertices = 'inclusiveCandidateNegativeSecondaryVertices',
0053 )
0054 pfNegativeParticleTransformerAK4JetTags = pfParticleTransformerAK4JetTags.clone(
0055     src = 'pfNegativeParticleTransformerAK4TagInfos',
0056 )
0057
0058 # probs
0059 _pfNegativeParticleTransformerAK4JetTagsProbs = ['pfNegativeParticleTransformerAK4JetTags:' + flav_name
0060                                  for flav_name in pfParticleTransformerAK4JetTags.flav_names]