TracksterInferenceByANN.cc

Go to the documentation of this file.

#include "RecoHGCal/TICL/interface/TracksterInferenceAlgoFactory.h"
#include "RecoHGCal/TICL/interface/TracksterInferenceByANN.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

namespace ticl {

  TracksterInferenceByANN::TracksterInferenceByANN(const edm::ParameterSet& conf) : TracksterInferenceAlgoBase(conf) {
    // Load ANN model
  }

  void TracksterInferenceByANN::inputData(const std::vector<reco::CaloCluster>& layerClusters,
                                          std::vector<Trackster>& tracksters) {
    // Prepare data for inference
  }

  void TracksterInferenceByANN::runInference(std::vector<Trackster>& tracksters) {
    // Run inference using ANN
  }
}  // namespace ticl

// Define this as a plug-in
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_EDM_PLUGIN(TracksterInferenceAlgoFactory, ticl::TracksterInferenceByANN, "TracksterInferenceByANN");