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
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
0010 # declare all the discriminators
0011 # probs
0012 _pfUnifiedParticleTransformerAK4JetTagsProbs = ['pfUnifiedParticleTransformerAK4JetTags:' + flav_name
0013                                  for flav_name in pfUnifiedParticleTransformerAK4JetTags.flav_names]
0014 # meta-taggers
0015 _pfUnifiedParticleTransformerAK4JetTagsMetaDiscrs = ['pfUnifiedParticleTransformerAK4DiscriminatorsJetTags:' + disc.name.value()
0016                                       for disc in pfUnifiedParticleTransformerAK4DiscriminatorsJetTags.discriminators]
0017 _pfUnifiedParticleTransformerAK4JetTagsAll = _pfUnifiedParticleTransformerAK4JetTagsProbs + _pfUnifiedParticleTransformerAK4JetTagsMetaDiscrs
0018
0019
0020
0021 # ==
0022 # This task is not used, useful only if we run it from RECO jets (RECO/AOD)
0023 pfUnifiedParticleTransformerAK4Task = cms.Task(puppi, primaryVertexAssociation,
0024                              pfUnifiedParticleTransformerAK4TagInfos, pfUnifiedParticleTransformerAK4JetTags,
0025                              pfUnifiedParticleTransformerAK4DiscriminatorsJetTags)
0026 # run from MiniAOD instead
0027 pfUnifiedParticleTransformerAK4FromMiniAODTask = cms.Task(pfUnifiedParticleTransformerAK4TagInfos,
0028                              pfUnifiedParticleTransformerAK4JetTags,
0029                              pfUnifiedParticleTransformerAK4DiscriminatorsJetTags)