TICL/plugins/TracksterInferenceByANN.cc

0001 #include "RecoHGCal/TICL/interface/TracksterInferenceAlgoFactory.h"
0002 #include "RecoHGCal/TICL/interface/TracksterInferenceByANN.h"
0003 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0004 #include "FWCore/Framework/interface/ConsumesCollector.h"
0005
0006 namespace ticl {
0007
0008   TracksterInferenceByANN::TracksterInferenceByANN(const edm::ParameterSet& conf) : TracksterInferenceAlgoBase(conf) {
0009     // Load ANN model
0010   }
0011
0012   void TracksterInferenceByANN::inputData(const std::vector<reco::CaloCluster>& layerClusters,
0013                                           std::vector<Trackster>& tracksters) {
0014     // Prepare data for inference
0015   }
0016
0017   void TracksterInferenceByANN::runInference(std::vector<Trackster>& tracksters) {
0018     // Run inference using ANN
0019   }
0020 }  // namespace ticl
0021
0022 // Define this as a plug-in
0023 #include "FWCore/Framework/interface/MakerMacros.h"
0024 DEFINE_EDM_PLUGIN(TracksterInferenceAlgoFactory, ticl::TracksterInferenceByANN, "TracksterInferenceByANN");