ticl::TracksterInferenceByANN Class Reference

#include <TracksterInferenceByANN.h>

Inheritance diagram for ticl::TracksterInferenceByANN:

Public Member Functions
void	inputData (const std::vector< reco::CaloCluster > &layerClusters, std::vector< Trackster > &tracksters) override
void	runInference (std::vector< Trackster > &tracksters) override
	TracksterInferenceByANN (const edm::ParameterSet &conf)
Public Member Functions inherited from ticl::TracksterInferenceAlgoBase
	TracksterInferenceAlgoBase (const edm::ParameterSet &conf)
virtual	~TracksterInferenceAlgoBase ()

Private Attributes
const cms::Ort::ONNXRuntime *	onnxEnergySession_
const cms::Ort::ONNXRuntime *	onnxPIDSession_

Additional Inherited Members
Static Public Member Functions inherited from ticl::TracksterInferenceAlgoBase
static void	fillPSetDescription (edm::ParameterSetDescription &desc)
Protected Attributes inherited from ticl::TracksterInferenceAlgoBase
int	algo_verbosity_

Detailed Description

Definition at line 7 of file TracksterInferenceByANN.h.

Constructor & Destructor Documentation

TracksterInferenceByANN()

ticl::TracksterInferenceByANN::TracksterInferenceByANN ( const edm::ParameterSet &  conf )

explicit

Definition at line 8 of file TracksterInferenceByANN.cc.

                                                                              : TracksterInferenceAlgoBase(conf) {
    // Load ANN model
  }

Member Function Documentation

inputData()

void ticl::TracksterInferenceByANN::inputData ( const std::vector< reco::CaloCluster > &  layerClusters, std::vector< Trackster > &  tracksters  )

overridevirtual

Implements ticl::TracksterInferenceAlgoBase.

Definition at line 12 of file TracksterInferenceByANN.cc.

                                                                            {
    // Prepare data for inference
  }

runInference()

void ticl::TracksterInferenceByANN::runInference ( std::vector< Trackster > &  tracksters )

overridevirtual

Implements ticl::TracksterInferenceAlgoBase.

Definition at line 17 of file TracksterInferenceByANN.cc.

                                                                             {
    // Run inference using ANN
  }

Member Data Documentation

onnxEnergySession_

const cms::Ort::ONNXRuntime* ticl::TracksterInferenceByANN::onnxEnergySession_

private

Definition at line 15 of file TracksterInferenceByANN.h.

onnxPIDSession_

const cms::Ort::ONNXRuntime* ticl::TracksterInferenceByANN::onnxPIDSession_

private

Definition at line 14 of file TracksterInferenceByANN.h.