![]() |
ATLAS Offline Software
|
Implements edge classification by inferencing on an Interaction Network GNN. More...
#include "AthenaBaseComps/AthAlgTool.h"#include "FPGATrackSimObjects/FPGATrackSimGNNHit.h"#include "FPGATrackSimObjects/FPGATrackSimGNNEdge.h"#include "AthOnnxInterfaces/IOnnxRuntimeInferenceTool.h"#include <onnxruntime_cxx_api.h>Go to the source code of this file.
Classes | |
| class | FPGATrackSimGNNEdgeClassifierTool |
Implements edge classification by inferencing on an Interaction Network GNN.
This class implements the scoreEdges() function which will calculate a score for each FPGATrackSimGNNEdge using an Interaction Network GNN. Each edge will receive a score between 0 and 1 that indicates how likely the edge is to connect two hits that belong to the same particle/track. The default value of the edge score is -1, indicating that an edge has not been scored by the GNN yet.
Definition in file FPGATrackSimGNNEdgeClassifierTool.h.