Class GNNPrediction

java.lang.Object

org.jlab.rec.alert.AI.GNNPrediction

public final class GNNPrediction
extends Object

Orchestrates GNN-based track finding: builds the graph, runs the exported edge scorer, extracts tracks via connected components on edge scores thresholded at 0.1, and converts each node-set back into a TrackCandidate carrying per-superlayer Clusters so the downstream helix fit / Kalman stages can consume it. Components that include ATOF graph nodes yield AHDC_ATOF candidates with those ATOF hits attached; the rest are AHDC_ONLY.

Constructor Summary

Constructors

Constructor	Description
GNNPrediction()

Method Summary

Modifier and Type	Method	Description
ArrayList<TrackCandidate>	prediction(List<Hit> ahdcHits, List<AtofHitStub> atofHits, ModelTrackFindingGNN model)

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

GNNPrediction

public GNNPrediction()

Method Details

prediction

public ArrayList<TrackCandidate> prediction(List<Hit> ahdcHits,
 List<AtofHitStub> atofHits,
 ModelTrackFindingGNN model)