File indexing completed on 2024-11-25 02:29:16
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 cylinder_radius_sqr_split = cms.double(9),
0038 proj_distance_split = cms.double(5),
0039 track_time_quality_threshold = cms.double(0.5),
0040 min_num_lcs = cms.uint32(7),
0041 min_trackster_energy = cms.double(10),
0042 pca_quality_th = cms.double(0.85),
0043 dot_prod_th = cms.double(0.97),
0044 deltaRxy = cms.double(4),
0045 lower_boundary = cms.vdouble(
0046 10,
0047 150
0048 ),
0049 upper_boundary = cms.vdouble(
0050 3,
0051 70
0052 ),
0053 upper_distance_projective_sqr = cms.vdouble(
0054 40,
0055 60
0056 ),
0057 lower_distance_projective_sqr = cms.vdouble(
0058 10,
0059 30
0060 ),
0061 min_distance_z = cms.vdouble(
0062 35,
0063 35
0064 ),
0065 upper_distance_projective_sqr_closest_points = cms.vdouble(
0066 10,
0067 30
0068 ),
0069 lower_distance_projective_sqr_closest_points = cms.vdouble(
0070 10,
0071 30
0072 ),
0073 max_z_distance_closest_points = cms.vdouble(
0074 35,
0075 35
0076 ),
0077 cylinder_radius_sqr = cms.vdouble(
0078 10,
0079 20
0080 ),
0081 algo_verbosity = cms.int32(0),
0082 type = cms.string('Skeletons')
0083
0084 ),
0085 tracksters_collections = cms.VInputTag('ticlTrackstersCLUE3DHigh'),
0086 original_masks = cms.VInputTag('hgcalMergeLayerClusters:InitialLayerClustersMask'),
0087 layer_clusters = cms.InputTag('hgcalMergeLayerClusters'),
0088 layer_clustersTime = cms.InputTag('hgcalMergeLayerClusters', 'timeLayerCluster'),
0089 regressionAndPid = cms.bool(False),
0090 detector = cms.string('HGCAL'),
0091 propagator = cms.string('PropagatorWithMaterial'),
0092 inferenceAlgo = cms.string('TracksterInferenceByDNN'),
0093 mightGet = cms.optional.untracked.vstring
0094 )
0095 for a in args:
0096 mod.update_(a)
0097 mod.update_(kwargs)
0098 return mod
