ONNXRuntime/python/pfUnifiedParticleTransformerAK4V1_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoBTag.FeatureTools.pfUnifiedParticleTransformerAK4TagInfos_cfi import pfUnifiedParticleTransformerAK4TagInfos
0004 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4JetTags_cfi import pfUnifiedParticleTransformerAK4JetTags
0005 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4V1DiscriminatorsJetTags_cfi import pfUnifiedParticleTransformerAK4V1DiscriminatorsJetTags
0006
0007 #
0008 pfUnifiedParticleTransformerAK4V1TagInfos = pfUnifiedParticleTransformerAK4TagInfos.clone()
0009
0010 pfUnifiedParticleTransformerAK4V1JetTags = pfUnifiedParticleTransformerAK4JetTags.clone(
0011   src = "pfUnifiedParticleTransformerAK4V1TagInfos",
0012   model_path = cms.FileInPath('RecoBTag/Combined/data/UParTAK4/PUPPI/V00/UParTAK4.onnx'),
0013 )
0014
0015 # declare all the discriminators
0016 # probs
0017 _pfUnifiedParticleTransformerAK4V1JetTagsProbs = ['pfUnifiedParticleTransformerAK4V1JetTags:' + flav_name
0018                                  for flav_name in pfUnifiedParticleTransformerAK4V1JetTags.flav_names]
0019 # meta-taggers
0020 _pfUnifiedParticleTransformerAK4V1JetTagsMetaDiscrs = ['pfUnifiedParticleTransformerAK4V1DiscriminatorsJetTags:' + disc.name.value()
0021                                       for disc in pfUnifiedParticleTransformerAK4V1DiscriminatorsJetTags.discriminators]
0022
0023 _pfUnifiedParticleTransformerAK4V1JetTagsAll = _pfUnifiedParticleTransformerAK4V1JetTagsProbs + _pfUnifiedParticleTransformerAK4V1JetTagsMetaDiscrs
0024
0025 # run from MiniAOD instead
0026 pfUnifiedParticleTransformerAK4V1FromMiniAODTask = cms.Task(pfUnifiedParticleTransformerAK4V1TagInfos,
0027   pfUnifiedParticleTransformerAK4V1JetTags,
0028   pfUnifiedParticleTransformerAK4V1DiscriminatorsJetTags
0029 )