Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Public Types | Public Member Functions | List of all members
Acts::MLTrackClassifier Class Reference

#include <acts/blob/sPHENIX/Plugins/Onnx/include/Acts/Plugins/Onnx/MLTrackClassifier.hpp>

+ Inheritance diagram for Acts::MLTrackClassifier:
+ Collaboration diagram for Acts::MLTrackClassifier:

Public Types

enum  TrackLabels
 The labels for track quality. More...
 

Public Member Functions

TrackLabels predictTrackLabel (std::vector< float > &inputFeatures, double decisionThreshProb) const
 Predict the track label.
 
bool isDuplicate (std::vector< float > &inputFeatures, double decisionThreshProb) const
 Check if the predicted track label is 'duplicate'.
 
- Public Member Functions inherited from Acts::OnnxRuntimeBase
 OnnxRuntimeBase ()=default
 Default constructor.
 
 OnnxRuntimeBase (Ort::Env &env, const char *modelPath)
 Parametrized constructor.
 
 ~OnnxRuntimeBase ()=default
 Default destructor.
 
std::vector< float > runONNXInference (std::vector< float > &inputTensorValues) const
 Run the ONNX inference function.
 
std::vector< std::vector< float > > runONNXInference (NetworkBatchInput &inputTensorValues) const
 Run the ONNX inference function for a batch of input.
 
std::vector< std::vector
< std::vector< float > > > 		runONNXInferenceMultiOutput (NetworkBatchInput &inputTensorValues) const
 Run the multi-output ONNX inference function for a batch of input.
 

Detailed Description

Definition at line 19 of file MLTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 19 of file MLTrackClassifier.hpp

Member Enumeration Documentation

enum Acts::MLTrackClassifier::TrackLabels

The labels for track quality.

Definition at line 24 of file MLTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 24 of file MLTrackClassifier.hpp

Member Function Documentation

bool Acts::MLTrackClassifier::isDuplicate ( std::vector< float > &  inputFeatures,
double  decisionThreshProb 
) const

Check if the predicted track label is 'duplicate'.

Parameters
inputFeaturesThe vector of input features for the trajectory to be classified
decisionThreshProbThe probability threshold used to predict the track label
Returns
If the predicted track label is 'duplicate'

Definition at line 38 of file MLTrackClassifier.cpp.

View newest version in sPHENIX GitHub at line 38 of file MLTrackClassifier.cpp

References predictTrackLabel().

Referenced by runRecCKFTracks().

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

Acts::MLTrackClassifier::TrackLabels Acts::MLTrackClassifier::predictTrackLabel ( std::vector< float > &  inputFeatures,
double  decisionThreshProb 
) const

Predict the track label.

Parameters
inputFeaturesThe vector of input features for the trajectory to be classified
decisionThreshProbThe probability threshold used to predict the track label
Returns
The predicted track label of the trajectory

Definition at line 15 of file MLTrackClassifier.cpp.

View newest version in sPHENIX GitHub at line 15 of file MLTrackClassifier.cpp

References Acts::OnnxRuntimeBase::runONNXInference().

Referenced by isDuplicate().

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

The documentation for this class was generated from the following files: