File indexing completed on 2025-02-07 14:24:03
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def ParticleTransformerAK4ONNXJetTagsProducer(*args, **kwargs):
0004 mod = cms.EDProducer('ParticleTransformerAK4ONNXJetTagsProducer',
0005 src = cms.InputTag('pfParticleTransformerAK4TagInfos'),
0006 input_names = cms.vstring(
0007 'input_1',
0008 'input_2',
0009 'input_3',
0010 'input_4',
0011 'input_5',
0012 'input_6'
0013 ),
0014 model_path = cms.FileInPath('RecoBTag/Combined/data/RobustParTAK4/PUPPI/V00/modelfile/model.onnx'),
0015 output_names = cms.vstring('softmax'),
0016 flav_names = cms.vstring(
0017 'probb',
0018 'probbb',
0019 'problepb',
0020 'probc',
0021 'probuds',
0022 'probg'
0023 ),
0024 mightGet = cms.optional.untracked.vstring
0025 )
0026 for a in args:
0027 mod.update_(a)
0028 mod.update_(kwargs)
0029 return mod
