File indexing completed on 2025-06-04 22:35:53
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 jets = cms.InputTag('hltAK4PFPuppiJets'),
0024 produceValueMap = cms.untracked.bool(False),
0025 mightGet = cms.optional.untracked.vstring
0026 )
0027 for a in args:
0028 mod.update_(a)
0029 mod.update_(kwargs)
0030 return mod
