File indexing completed on 2024-04-06 12:00:18
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def HcalMLTask(**kwargs):
0004 mod = cms.EDProducer('HcalMLTask',
0005 name = cms.untracked.string('HcalMLTask'),
0006 onnx_model_path_HB = cms.untracked.string('DQM/HcalTasks/data/HB_2022/CGAE_MultiDim_SPATIAL_vONNX_RCLv22_PIXEL_BT_BN_RIN_IPHI_MED_5218_v06_02_2023_21h01_stateful.onnx'),
0007 onnx_model_path_HE = cms.untracked.string('DQM/HcalTasks/data/HE_2022/CGAE_MultiDim_SPATIAL_vONNX_RCLv22_PIXEL_BT_BN_RIN_IPHI_MED_7763_v06_02_2023_22h55_stateful.onnx'),
0008 flagDecisionThr = cms.untracked.double(20),
0009 debug = cms.untracked.int32(0),
0010 runkeyVal = cms.untracked.int32(0),
0011 runkeyName = cms.untracked.string('pp_run'),
0012 ptype = cms.untracked.int32(1),
0013 mtype = cms.untracked.bool(True),
0014 subsystem = cms.untracked.string('Hcal'),
0015 tagHBHE = cms.untracked.InputTag('hcalDigis'),
0016 tagHO = cms.untracked.InputTag('hcalDigis'),
0017 tagHF = cms.untracked.InputTag('hcalDigis'),
0018 mightGet = cms.optional.untracked.vstring
0019 )
0020 for k,v in kwargs.items():
0021 setattr(mod, k, v)
0022 return mod