File indexing completed on 2025-02-07 14:24:03
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def UnifiedParticleTransformerAK4ONNXJetTagsProducer(*args, **kwargs):
0004 mod = cms.EDProducer('UnifiedParticleTransformerAK4ONNXJetTagsProducer',
0005 src = cms.InputTag('pfUnifiedParticleTransformerAK4TagInfos'),
0006 input_names = cms.vstring(
0007 'input_1',
0008 'input_2',
0009 'input_3',
0010 'input_4',
0011 'input_5',
0012 'input_6',
0013 'input_7',
0014 'input_8'
0015 ),
0016 model_path = cms.FileInPath('RecoBTag/Combined/data/UParTAK4/PUPPI/V01/UParTAK4_v2.onnx'),
0017 output_names = cms.vstring('softmax'),
0018 flav_names = cms.vstring(
0019 'probb',
0020 'probbb',
0021 'problepb',
0022 'probc',
0023 'probs',
0024 'probu',
0025 'probd',
0026 'probg',
0027 'probele',
0028 'probmu',
0029 'probtaup1h0p',
0030 'probtaup1h1p',
0031 'probtaup1h2p',
0032 'probtaup3h0p',
0033 'probtaup3h1p',
0034 'probtaum1h0p',
0035 'probtaum1h1p',
0036 'probtaum1h2p',
0037 'probtaum3h0p',
0038 'probtaum3h1p',
0039 'ptcorr',
0040 'ptreshigh',
0041 'ptreslow',
0042 'ptnu',
0043 'probemudata',
0044 'probemumc',
0045 'probdimudata',
0046 'probdimumc',
0047 'probmutaudata',
0048 'probmutaumc'
0049 ),
0050 mightGet = cms.optional.untracked.vstring
0051 )
0052 for a in args:
0053 mod.update_(a)
0054 mod.update_(kwargs)
0055 return mod
