Class Acts::TorchMetricLearning

class TorchMetricLearning : public Acts::GraphConstructionBase

Public Functions

TorchMetricLearning(const Config &cfg, std::unique_ptr<const Logger> logger)

~TorchMetricLearning()

inline Config config() const

virtual std::tuple<std::any, std::any> operator()(std::vector<float> &inputValues) override

Perform the graph construction.

Parameters

inputValues – Flattened input data

Returns

(node_tensor, edge_tensore)

struct Config

Public Members

int embeddingDim = 8

int knnVal = 500

std::string modelPath

float rVal = 1.6

int spacepointFeatures = 3