File MLTrackClassifier.hpp

namespace Acts

Note

This file is foreseen for the Geometry module to replace Extent

class MLTrackClassifier : public Acts::OnnxRuntimeBase
#include </home/docs/checkouts/readthedocs.org/user_builds/acts/checkouts/v20.0.0/Plugins/Onnx/include/Acts/Plugins/Onnx/MLTrackClassifier.hpp>

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