Back to home page

Project CMSSW displayed by LXR

 
 

    


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