ONNXRuntime/python/pfUnifiedParticleTransformerAK4_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoBTag.FeatureTools.pfUnifiedParticleTransformerAK4TagInfos_cfi import pfUnifiedParticleTransformerAK4TagInfos
0004
0005 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4JetTags_cfi import pfUnifiedParticleTransformerAK4JetTags as _pfUnifiedParticleTransformerAK4JetTags
0006 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4DiscriminatorsJetTags_cfi import pfUnifiedParticleTransformerAK4DiscriminatorsJetTags
0007 from CommonTools.PileupAlgos.Puppi_cff import puppi
0008 from CommonTools.RecoAlgos.primaryVertexAssociation_cfi import primaryVertexAssociation
0009 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4SonicJetTags_cfi import pfUnifiedParticleTransformerAK4SonicJetTags as _pfUnifiedParticleTransformerAK4SonicJetTags
0010 from Configuration.ProcessModifiers.unifiedparticleTransformerAK4SonicTriton_cff import unifiedparticleTransformerAK4SonicTriton
0011
0012 pfUnifiedParticleTransformerAK4JetTags = _pfUnifiedParticleTransformerAK4JetTags.clone()
0013
0014 unifiedparticleTransformerAK4SonicTriton.toReplaceWith(pfUnifiedParticleTransformerAK4JetTags, _pfUnifiedParticleTransformerAK4SonicJetTags.clone(
0015     Client = cms.PSet(
0016         timeout = cms.untracked.uint32(500),
0017         mode = cms.string("Async"),
0018         modelName = cms.string("unifiedparticletransformer_AK4_V01"),
0019         modelConfigPath = cms.FileInPath("RecoBTag/Combined/data/models/unifiedparticletransformer_AK4_V01/config.pbtxt"),
0020         modelVersion = cms.string(""),
0021         verbose = cms.untracked.bool(True),
0022         allowedTries = cms.untracked.uint32(0),
0023         useSharedMemory = cms.untracked.bool(True),
0024         compression = cms.untracked.string(""),
0025     ),
0026     flav_names = pfUnifiedParticleTransformerAK4JetTags.flav_names,
0027 ))
0028
0029 # declare all the discriminators
0030 # probs
0031 _pfUnifiedParticleTransformerAK4JetTagsProbs = ['pfUnifiedParticleTransformerAK4JetTags:' + flav_name
0032                                  for flav_name in pfUnifiedParticleTransformerAK4JetTags.flav_names]
0033 # meta-taggers
0034 _pfUnifiedParticleTransformerAK4JetTagsMetaDiscrs = ['pfUnifiedParticleTransformerAK4DiscriminatorsJetTags:' + disc.name.value()
0035                                       for disc in pfUnifiedParticleTransformerAK4DiscriminatorsJetTags.discriminators]
0036 _pfUnifiedParticleTransformerAK4JetTagsAll = _pfUnifiedParticleTransformerAK4JetTagsProbs + _pfUnifiedParticleTransformerAK4JetTagsMetaDiscrs
0037
0038
0039
0040 # ==
0041 # This task is not used, useful only if we run it from RECO jets (RECO/AOD)
0042 pfUnifiedParticleTransformerAK4Task = cms.Task(puppi, primaryVertexAssociation,
0043                              pfUnifiedParticleTransformerAK4TagInfos, pfUnifiedParticleTransformerAK4JetTags,
0044                              pfUnifiedParticleTransformerAK4DiscriminatorsJetTags)
0045 # run from MiniAOD instead
0046 pfUnifiedParticleTransformerAK4FromMiniAODTask = cms.Task(pfUnifiedParticleTransformerAK4TagInfos,
0047                              pfUnifiedParticleTransformerAK4JetTags,
0048                              pfUnifiedParticleTransformerAK4DiscriminatorsJetTags)
0049
0050 # === Negative tags ===
0051 pfNegativeUnifiedParticleTransformerAK4TagInfos = pfUnifiedParticleTransformerAK4TagInfos.clone(
0052     flip = True,
0053     secondary_vertices = 'inclusiveCandidateNegativeSecondaryVertices',
0054 )
0055 pfNegativeUnifiedParticleTransformerAK4JetTags = pfUnifiedParticleTransformerAK4JetTags.clone(
0056     src = 'pfNegativeParticleTransformerAK4TagInfos',
0057 )
0058
0059 # probs
0060 _pfNegativeUnifiedParticleTransformerAK4JetTagsProbs = ['pfNegativeUnifiedParticleTransformerAK4JetTags:' + flav_name
0061                                  for flav_name in pfUnifiedParticleTransformerAK4JetTags.flav_names]