File indexing completed on 2024-09-26 05:06:14
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def DeepVertexONNXJetTagsProducer(*args, **kwargs):
0004 mod = cms.EDProducer('DeepVertexONNXJetTagsProducer',
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 'input_6',
0013 'input_7',
0014 'input_8',
0015 'input_9',
0016 'input_10',
0017 'input_11',
0018 'input_12'
0019 ),
0020 model_path = cms.FileInPath('RecoBTag/Combined/data/DeepVertex/phase1_deepvertex.onnx'),
0021 output_names = cms.vstring('dense_6'),
0022 flav_names = cms.vstring(
0023 'probb',
0024 'probc',
0025 'probuds',
0026 'probg'
0027 ),
0028 min_jet_pt = cms.double(15),
0029 max_jet_eta = cms.double(2.5),
0030 mightGet = cms.optional.untracked.vstring
0031 )
0032 for a in args:
0033 mod.update_(a)
0034 mod.update_(kwargs)
0035 return mod
