@@ -74,7 +74,7 @@ namespace ticl {
7474 input_Data_[1 ][index_tr + 3 ] = static_cast <float >(trackster.barycenter ().y ());
7575 input_Data_[1 ][index_tr + 4 ] = static_cast <float >(std::abs (trackster.barycenter ().z ()));
7676 input_Data_[1 ][index_tr + 5 ] = static_cast <float >(std::abs (trackster.barycenter ().eta ()));
77- input_Data_[1 ][index_tr + 6 ] = static_cast <float >(trackster.barycenter ().phi ());
77+ input_Data_[1 ][index_tr + 6 ] = static_cast <float >(trackster.barycenter ().phi ());
7878
7979 // Prepare indices and sort clusters based on energy
8080 std::vector<int > clusterIndices (trackster.vertices ().size ());
@@ -85,7 +85,7 @@ namespace ticl {
8585 std::sort (clusterIndices.begin (), clusterIndices.end (), [&layerClusters, &trackster](const int & a, const int & b) {
8686 return layerClusters[trackster.vertices (a)].energy () > layerClusters[trackster.vertices (b)].energy ();
8787 });
88-
88+
8989 std::vector<int > seenClusters (eidNLayers_, 0 );
9090
9191 // Fill input data with cluster information
@@ -94,7 +94,7 @@ namespace ticl {
9494 int j = rhtools_.getLayerWithOffset (cluster.hitsAndFractions ()[0 ].first ) - 1 ;
9595 if (j < eidNLayers_ && seenClusters[j] < eidNClusters_) {
9696 auto index_lc = (i * eidNLayers_ + j) * eidNFeatures_ * eidNClusters_ + seenClusters[j] * eidNFeatures_;
97- // Adding more features regarding LC, such as E, eta, phi, x, y, z, and nhits.
97+ // Adding more features regarding LC, such as E, eta, phi, x, y, z, and nhits.
9898 input_Data_[0 ][index_lc] =
9999 static_cast <float >(cluster.energy () / static_cast <float >(trackster.vertex_multiplicity (k)));
100100 input_Data_[0 ][index_lc + 1 ] = static_cast <float >(std::abs (cluster.eta ()));
@@ -142,15 +142,16 @@ namespace ticl {
142142 void TracksterInferenceByPFN::fillPSetDescription (edm::ParameterSetDescription& iDesc) {
143143 iDesc.add <int >(" algo_verbosity" , 0 );
144144 iDesc
145- .add <edm::FileInPath>(" onnxPIDModelPath" ,
146- edm::FileInPath (" RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/id_v0.onnx" ))
145+ .add <edm::FileInPath>(
146+ " onnxPIDModelPath" ,
147+ edm::FileInPath (" RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/id_v0.onnx" ))
147148 ->setComment (" Path to ONNX PID model CLU3D" );
148149 iDesc
149150 .add <edm::FileInPath>(
150151 " onnxEnergyModelPath" ,
151152 edm::FileInPath (" RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/energy_v0.onnx" ))
152153 ->setComment (" Path to ONNX Energy model CLU3D" );
153- iDesc.add <std::vector<std::string>>(" inputNames" , {" input" ," input_tr_features" });
154+ iDesc.add <std::vector<std::string>>(" inputNames" , {" input" , " input_tr_features" });
154155 iDesc.add <std::vector<std::string>>(" output_en" , {" enreg_output" });
155156 iDesc.add <std::vector<std::string>>(" output_id" , {" pid_output" });
156157 iDesc.add <double >(" eid_min_cluster_energy" , 1.0 );
0 commit comments