File OnnxMetricLearning.hpp¶
-
namespace Acts
Note
This file is foreseen for the
Geometry
module to replaceExtent
-
class OnnxMetricLearning : public Acts::GraphConstructionBase
- #include <Acts/Plugins/ExaTrkX/OnnxMetricLearning.hpp>
Public Functions
-
OnnxMetricLearning(const Config &cfg)
-
~OnnxMetricLearning()
-
inline Config config() const
-
virtual std::tuple<std::any, std::any> operator()(std::vector<float> &inputValues, const Logger &logger) override
Perform the graph construction.
- Parameters
inputValues – Flattened input data
logger – Logger instance
- Returns
(node_tensor, edge_tensore)
Private Functions
Private Members
-
struct Config
- #include <Acts/Plugins/ExaTrkX/OnnxMetricLearning.hpp>
Public Members
-
int embeddingDim = 8
-
int knnVal = 500
-
std::string modelPath
-
float rVal = 1.6
-
int spacepointFeatures = 3
-
int embeddingDim = 8
-
OnnxMetricLearning(const Config &cfg)
-
class OnnxMetricLearning : public Acts::GraphConstructionBase
-
namespace Ort