Class Acts::MLTrackClassifier
-
class MLTrackClassifier : public Acts::OnnxRuntimeBase
Public Types
Public Functions
-
bool isDuplicate(std::vector<float> &inputFeatures, double decisionThreshProb) const
Check if the predicted track label is ‘duplicate’.
- Parameters
inputFeatures – The vector of input features for the trajectory to be classified
decisionThreshProb – The probability threshold used to predict the track label
- Returns
If the predicted track label is ‘duplicate’
-
TrackLabels predictTrackLabel(std::vector<float> &inputFeatures, double decisionThreshProb) const
Predict the track label.
- Parameters
inputFeatures – The vector of input features for the trajectory to be classified
decisionThreshProb – The probability threshold used to predict the track label
- Returns
The predicted track label of the trajectory
-
bool isDuplicate(std::vector<float> &inputFeatures, double decisionThreshProb) const