Class Acts::MLTrackClassifier

class Acts::MLTrackClassifier : public Acts::OnnxRuntimeBase

Public Types

enum TrackLabels

The labels for track quality.

Values:

enumerator eGood
enumerator eDuplicate
enumerator eFake

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