File indexing completed on 2024-09-26 05:06:15
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def TracksterLinksProducer(*args, **kwargs):
0004 mod = cms.EDProducer('TracksterLinksProducer',
0005 pluginInferenceAlgoTracksterInferenceByDNN = cms.PSet(
0006 algo_verbosity = cms.int32(0),
0007 onnxPIDModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/patternrecognition/id_v0.onnx'),
0008 onnxEnergyModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/patternrecognition/energy_v0.onnx'),
0009 inputNames = cms.vstring('input'),
0010 output_en = cms.vstring('enreg_output'),
0011 output_id = cms.vstring('pid_output'),
0012 eid_min_cluster_energy = cms.double(1),
0013 eid_n_layers = cms.int32(50),
0014 eid_n_clusters = cms.int32(10),
0015 doPID = cms.int32(1),
0016 doRegression = cms.int32(1),
0017 type = cms.string('TracksterInferenceByDNN')
0018
0019 ),
0020 pluginInferenceAlgoTracksterInferenceByCNNv4 = cms.PSet(
0021 algo_verbosity = cms.int32(0),
0022 onnxModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv4/onnx_models/energy_id_v0.onnx'),
0023 inputNames = cms.vstring('input:0'),
0024 outputNames = cms.vstring(
0025 'output/regressed_energy:0',
0026 'output/id_probabilities:0'
0027 ),
0028 eid_min_cluster_energy = cms.double(1),
0029 eid_n_layers = cms.int32(50),
0030 eid_n_clusters = cms.int32(10),
0031 doPID = cms.int32(1),
0032 doRegression = cms.int32(0),
0033 type = cms.string('TracksterInferenceByCNNv4')
0034
0035 ),
0036 linkingPSet = cms.PSet(
0037 track_time_quality_threshold = cms.double(0.5),
0038 wind = cms.double(0.036),
0039 min_num_lcs = cms.uint32(7),
0040 min_trackster_energy = cms.double(10),
0041 pca_quality_th = cms.double(0.85),
0042 dot_prod_th = cms.double(0.97),
0043 max_distance_projective_sqr = cms.vdouble(
0044 60,
0045 60
0046 ),
0047 min_distance_z = cms.vdouble(
0048 30,
0049 30
0050 ),
0051 max_distance_projective_sqr_closest_points = cms.vdouble(
0052 60,
0053 60
0054 ),
0055 max_z_distance_closest_points = cms.vdouble(
0056 35,
0057 35
0058 ),
0059 cylinder_radius_sqr = cms.vdouble(
0060 9,
0061 9
0062 ),
0063 algo_verbosity = cms.int32(0),
0064 type = cms.string('Skeletons')
0065
0066 ),
0067 tracksters_collections = cms.VInputTag('ticlTrackstersCLUE3DHigh'),
0068 original_masks = cms.VInputTag('hgcalMergeLayerClusters:InitialLayerClustersMask'),
0069 layer_clusters = cms.InputTag('hgcalMergeLayerClusters'),
0070 layer_clustersTime = cms.InputTag('hgcalMergeLayerClusters', 'timeLayerCluster'),
0071 regressionAndPid = cms.bool(False),
0072 detector = cms.string('HGCAL'),
0073 propagator = cms.string('PropagatorWithMaterial'),
0074 inferenceAlgo = cms.string('TracksterInferenceByDNN'),
0075 mightGet = cms.optional.untracked.vstring
0076 )
0077 for a in args:
0078 mod.update_(a)
0079 mod.update_(kwargs)
0080 return mod
