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