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¶