InDet::SiGNNTrackFinderTool Class Reference

InDet::SiGNNTrackFinderTool is a tool that produces track candidates with graph neural networks-based pipeline using 3D space points as inputs. More...

#include <SiGNNTrackFinderTool.h>

Inheritance diagram for InDet::SiGNNTrackFinderTool:

Collaboration diagram for InDet::SiGNNTrackFinderTool:

Public Member Functions
	SiGNNTrackFinderTool (const std::string &type, const std::string &name, const IInterface *parent)
virtual StatusCode	initialize () override
virtual StatusCode	getTracks (const std::vector< const Trk::SpacePoint * > &spacepoints, std::vector< std::vector< uint32_t > > &tracks) const override
	Get track candidates from a list of space points.
virtual MsgStream &	dump (MsgStream &out) const override
virtual std::ostream &	dump (std::ostream &out) const override

Protected Member Functions
	SiGNNTrackFinderTool ()=delete
	SiGNNTrackFinderTool (const SiGNNTrackFinderTool &)=delete
SiGNNTrackFinderTool &	operator= (const SiGNNTrackFinderTool &)=delete

Exa.TrkX pipeline configurations, which will not be changed after construction
StringProperty	m_inputMLModuleDir {this, "inputMLModelDir", ""}
UnsignedIntegerProperty	m_embeddingDim {this, "embeddingDim", 8}
FloatProperty	m_rVal {this, "rVal", 0.12}
UnsignedIntegerProperty	m_knnVal {this, "knnVal", 1000}
FloatProperty	m_filterCut {this, "filterCut", 0.05}
FloatProperty	m_ccCut {this, "ccCut", 0.01}
FloatProperty	m_walkMin {this, "walkMin", 0.1}
FloatProperty	m_walkMax {this, "walkMax", 0.6}
StringProperty	m_embeddingFeatureNames
StringProperty	m_embeddingFeatureScales
StringProperty	m_filterFeatureNames
StringProperty	m_filterFeatureScales
StringProperty	m_gnnFeatureNames
StringProperty	m_gnnFeatureScales
ToolHandle< AthOnnx::IOnnxRuntimeInferenceTool >	m_embedSessionTool
ToolHandle< AthOnnx::IOnnxRuntimeInferenceTool >	m_filterSessionTool
ToolHandle< AthOnnx::IOnnxRuntimeInferenceTool >	m_gnnSessionTool
ToolHandle< ISpacepointFeatureTool >	m_spacepointFeatureTool
std::vector< std::string >	m_embeddingFeatureNamesVec
std::vector< float >	m_embeddingFeatureScalesVec
std::vector< std::string >	m_filterFeatureNamesVec
std::vector< float >	m_filterFeatureScalesVec
std::vector< std::string >	m_gnnFeatureNamesVec
std::vector< float >	m_gnnFeatureScalesVec
void	initTrainedModels ()
MsgStream &	dumpevent (MsgStream &out) const

Detailed Description

InDet::SiGNNTrackFinderTool is a tool that produces track candidates with graph neural networks-based pipeline using 3D space points as inputs.

Author

xiang.nosp@m.yang.nosp@m..ju@c.nosp@m.ern..nosp@m.ch

Definition at line 31 of file SiGNNTrackFinderTool.h.

Constructor & Destructor Documentation

SiGNNTrackFinderTool() [1/3]

InDet::SiGNNTrackFinderTool::SiGNNTrackFinderTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent )

Definition at line 15 of file SiGNNTrackFinderTool.cxx.

                                                                         :
    base_class(type, name, parent)
  {
    declareInterface<IGNNTrackFinder>(this);
  }

SiGNNTrackFinderTool() [2/3]

InDet::SiGNNTrackFinderTool::SiGNNTrackFinderTool ( )

protecteddelete

SiGNNTrackFinderTool() [3/3]

InDet::SiGNNTrackFinderTool::SiGNNTrackFinderTool ( const SiGNNTrackFinderTool & )

protecteddelete

Member Function Documentation

dump() [1/2]

MsgStream & InDet::SiGNNTrackFinderTool::dump ( MsgStream & out ) const

overridevirtual

Definition at line 62 of file SiGNNTrackFinderTool.cxx.

{
  out<<std::endl;
  return dumpevent(out);
}

dump() [2/2]

std::ostream & InDet::SiGNNTrackFinderTool::dump ( std::ostream & out ) const

overridevirtual

Definition at line 68 of file SiGNNTrackFinderTool.cxx.

{
  return out;
}

dumpevent()

MsgStream & InDet::SiGNNTrackFinderTool::dumpevent ( MsgStream & out ) const

protected

Definition at line 73 of file SiGNNTrackFinderTool.cxx.

{
  out<<"|---------------------------------------------------------------------|"
       <<std::endl;
  out<<"| Number output tracks    | "<<std::setw(12)  
     <<"                              |"<<std::endl;
  out<<"|---------------------------------------------------------------------|"
     <<std::endl;
  return out;
}

getTracks()

StatusCode InDet::SiGNNTrackFinderTool::getTracks ( const std::vector< const Trk::SpacePoint * > & spacepoints, std::vector< std::vector< uint32_t > > & tracks ) const

overridevirtual

Get track candidates from a list of space points.

Parameters

spacepoints	a list of spacepoints as inputs to the GNN-based track finder.
tracks	a list of track candidates.

Returns

Definition at line 84 of file SiGNNTrackFinderTool.cxx.

{
  int64_t numSpacepoints = (int64_t)spacepoints.size();
  std::vector<float> eNodeFeatures;
  std::vector<float> fNodeFeatures;
  std::vector<float> gNodeFeatures;
  std::vector<uint32_t> spacepointIDs;
  std::vector<int> regions;
 
  int sp_idx = 0;
  for(const auto& sp: spacepoints){
    auto featureMap = m_spacepointFeatureTool->getFeatures(sp);
    regions.push_back(featureMap["region"]);
    // fill embedding node features.
    for(size_t i = 0; i < m_embeddingFeatureNamesVec.size(); i++){
      eNodeFeatures.push_back(
        featureMap[m_embeddingFeatureNamesVec[i]] / m_embeddingFeatureScalesVec[i]);
    }
 
    // fill filtering node features.
    for(size_t i = 0; i < m_filterFeatureNamesVec.size(); i++){
      fNodeFeatures.push_back(
        featureMap[m_filterFeatureNamesVec[i]] / m_filterFeatureScalesVec[i]);
    }
 
    // fill gnn node features.
    for(size_t i = 0; i < m_gnnFeatureNamesVec.size(); i++){
      gNodeFeatures.push_back(
        featureMap[m_gnnFeatureNamesVec[i]] / m_gnnFeatureScalesVec[i]);
    }
 
    spacepointIDs.push_back(sp_idx++);
  }
    // ************
    // Embedding
    // ************
    std::vector<int64_t> eInputShape{numSpacepoints, (long int) m_embeddingFeatureNamesVec.size()};
    std::vector<Ort::Value> eInputTensor;
    ATH_CHECK( m_embedSessionTool->addInput(eInputTensor, eNodeFeatures, 0, numSpacepoints) );
 
    std::vector<Ort::Value> eOutputTensor;
    std::vector<float> eOutputData;
    ATH_CHECK( m_embedSessionTool->addOutput(eOutputTensor, eOutputData, 0, numSpacepoints) );
 
    ATH_CHECK( m_embedSessionTool->inference(eInputTensor, eOutputTensor) );
 
    // ************
    // Building Edges
    // ************
    std::vector<int64_t> senders;
    std::vector<int64_t> receivers;
    ExaTrkXUtils::buildEdges(eOutputData, senders, receivers, numSpacepoints, m_embeddingDim, m_rVal, m_knnVal);
    int64_t numEdges = senders.size();
 
    // clean up embedding data.
    eNodeFeatures.clear();
    eInputTensor.clear();
    eOutputData.clear();
    eOutputTensor.clear();
 
    // sort the edge list and remove duplicate edges.
    std::vector<std::pair<int64_t, int64_t>> edgePairs;
    for(int64_t idx = 0; idx < numEdges; idx ++ ) {
        edgePairs.push_back({senders[idx], receivers[idx]});
    }
    std::sort(edgePairs.begin(), edgePairs.end());
    edgePairs.erase(std::unique(edgePairs.begin(), edgePairs.end()), edgePairs.end());
    
    // random shuffle the edge list.
    std::random_device rd;
    std::mt19937 rdm_gen(rd());
    std::random_shuffle(edgePairs.begin(), edgePairs.end());
 
    // sort the edge list by the sender * numSpacepoints + receiver.
    std::sort(edgePairs.begin(), edgePairs.end(), 
        [numSpacepoints](const std::pair<int64_t, int64_t>& a, const std::pair<int64_t, int64_t>& b){
            return a.first * numSpacepoints + a.second < b.first * numSpacepoints + b.second;
    });
 
    // convert the edge list to senders and receivers.
    senders.clear();
    receivers.clear();
    for(const auto& edge: edgePairs){
        senders.push_back(edge.first);
        receivers.push_back(edge.second);
    }
 
    edgePairs.clear();
 
    // ************
    // Filtering
    // ************
    std::vector<Ort::Value> fInputTensor;
    ATH_CHECK( m_filterSessionTool->addInput(fInputTensor, fNodeFeatures, 0, numSpacepoints) );
 
    std::vector<int64_t> edgeList(numEdges * 2);
    std::copy(senders.begin(), senders.end(), edgeList.begin());
    std::copy(receivers.begin(), receivers.end(), edgeList.begin() + senders.size());
 
    
    ATH_CHECK( m_filterSessionTool->addInput(fInputTensor, edgeList, 1, numEdges) );
 
    std::vector<float> fOutputData;
    std::vector<Ort::Value> fOutputTensor;
    ATH_CHECK( m_filterSessionTool->addOutput(fOutputTensor, fOutputData, 0, numEdges) );
 
    ATH_CHECK( m_filterSessionTool->inference(fInputTensor, fOutputTensor) );
 
    // apply sigmoid to the filtering output data
    // and remove edges with score < filterCut
    // and sort the edge list so that sender idx < receiver.
    std::vector<int64_t> rowIndices;
    std::vector<int64_t> colIndices;
    for (int64_t i = 0; i < numEdges; i++){
        float v = 1.f / (1.f + std::exp(-fOutputData[i]));  // sigmoid, float type
        if (v >= m_filterCut){
            auto src = edgeList[i];
            auto dst = edgeList[numEdges + i];
            if (src > dst) {
                std::swap(src, dst);
            }
            rowIndices.push_back(src);
            colIndices.push_back(dst);
        };
    };
    int64_t numEdgesAfterF = rowIndices.size();
 
    // clean up filtering data.
    fNodeFeatures.clear();
    fInputTensor.clear();
    fOutputData.clear();
    fOutputTensor.clear();
    // clean up sender and receiver list.
    senders.clear();
    receivers.clear();
 
    std::vector<int64_t> edgesAfterFiltering(numEdgesAfterF * 2);
    std::copy(rowIndices.begin(), rowIndices.end(), edgesAfterFiltering.begin());
    std::copy(colIndices.begin(), colIndices.end(), edgesAfterFiltering.begin() + senders.size());
 
    // ************
    // GNN
    // ************
 
    // use the same features for regions (2, 6)
    for(size_t idx = 0; idx < static_cast<size_t>(numSpacepoints); idx++){
        if (regions[idx] == 2 || regions[idx] == 6){
          for(size_t i = 4; i < m_gnnFeatureNamesVec.size(); i++){
              gNodeFeatures[idx * m_gnnFeatureNamesVec.size() + i] = gNodeFeatures[idx * m_gnnFeatureNamesVec.size() + i % 4];
          }
        }
    }
  
    std::vector<Ort::Value> gInputTensor;
    ATH_CHECK( m_gnnSessionTool->addInput(gInputTensor, gNodeFeatures, 0, numSpacepoints) );
    ATH_CHECK( m_gnnSessionTool->addInput(gInputTensor, edgesAfterFiltering, 1, numEdgesAfterF) );
 
    // calculate the edge features.
    std::vector<float> gnnEdgeFeatures;
    ExaTrkXUtils::calculateEdgeFeatures(gNodeFeatures, numSpacepoints, rowIndices, colIndices, gnnEdgeFeatures);
    ATH_CHECK( m_gnnSessionTool->addInput(gInputTensor, gnnEdgeFeatures, 2, numEdgesAfterF) );
    
    // gnn outputs
    std::vector<float> gOutputData;
    std::vector<Ort::Value> gOutputTensor;
    ATH_CHECK( m_gnnSessionTool->addOutput(gOutputTensor, gOutputData, 0, numEdgesAfterF) );
 
    ATH_CHECK( m_gnnSessionTool->inference(gInputTensor, gOutputTensor) );
    // apply sigmoid to the gnn output data
    for(auto& v : gOutputData){
        v = 1.f / (1.f + std::exp(-v));
    };
 
    // clean up GNN data.
    gNodeFeatures.clear();
    gInputTensor.clear();
    edgesAfterFiltering.clear();
 
    // ************
    // Track Labeling with cugraph::connected_components
    // ************
    tracks.clear();
    ExaTrkXUtils::CCandWalk(
      numSpacepoints,
      rowIndices, colIndices, gOutputData,
      tracks, m_ccCut, m_walkMin, m_walkMax
    );
 
    return StatusCode::SUCCESS;
}

initialize()

StatusCode InDet::SiGNNTrackFinderTool::initialize ( )

overridevirtual

Definition at line 22 of file SiGNNTrackFinderTool.cxx.

                                                 {
  ATH_CHECK( m_embedSessionTool.retrieve() );
  m_embedSessionTool->printModelInfo();
 
  ATH_CHECK( m_filterSessionTool.retrieve() );
  m_filterSessionTool->printModelInfo();
 
  ATH_CHECK( m_gnnSessionTool.retrieve() );
  m_gnnSessionTool->printModelInfo();
 
  // tokenize the feature names by comma and push to the vector
  auto split_fn = [](const std::string& s, auto convert_fn) {
      using ReturnType = std::decay_t<decltype(convert_fn(std::declval<std::string>()))>;
      std::vector<ReturnType> tokens;
      std::string token;
      std::istringstream tokenStream(s);
      while (std::getline(tokenStream, token, ',')) {
          token = token.substr(token.find_first_not_of(" "), token.find_last_not_of(" ") + 1);
          tokens.push_back(convert_fn(token));
      }
      return tokens;
  };
  auto convert_to_float = [](const std::string& s) -> float { return std::stof(s); };
  auto convert_to_str = [](const std::string& s) -> std::string { return s; };
 
  m_embeddingFeatureNamesVec = split_fn(m_embeddingFeatureNames, convert_to_str);
  m_embeddingFeatureScalesVec = split_fn(m_embeddingFeatureScales, convert_to_float);
  assert(m_embeddingFeatureNamesVec.size() == m_embeddingFeatureScalesVec.size());
 
  m_filterFeatureNamesVec = split_fn(m_filterFeatureNames, convert_to_str);
  m_filterFeatureScalesVec = split_fn(m_filterFeatureScales, convert_to_float);
  assert(m_filterFeatureNamesVec.size() == m_filterFeatureScalesVec.size());
 
  m_gnnFeatureNamesVec = split_fn(m_gnnFeatureNames, convert_to_str);
  m_gnnFeatureScalesVec = split_fn(m_gnnFeatureScales, convert_to_float);
  assert(m_gnnFeatureNamesVec.size() == m_gnnFeatureScalesVec.size());
 
  return StatusCode::SUCCESS;
}

initTrainedModels()

void InDet::SiGNNTrackFinderTool::initTrainedModels ( )

protected

operator=()

SiGNNTrackFinderTool & InDet::SiGNNTrackFinderTool::operator= ( const SiGNNTrackFinderTool & )

protecteddelete

Member Data Documentation

m_ccCut

FloatProperty InDet::SiGNNTrackFinderTool::m_ccCut {this, "ccCut", 0.01}

protected

Definition at line 70 of file SiGNNTrackFinderTool.h.

{this, "ccCut", 0.01};

m_embeddingDim

UnsignedIntegerProperty InDet::SiGNNTrackFinderTool::m_embeddingDim {this, "embeddingDim", 8}

protected

Definition at line 66 of file SiGNNTrackFinderTool.h.

{this, "embeddingDim", 8};

m_embeddingFeatureNames

StringProperty InDet::SiGNNTrackFinderTool::m_embeddingFeatureNames

protected

Initial value:

{
      this, "EmbeddingFeatureNames",
      "r, phi, z, cluster_x_1, cluster_y_1, cluster_z_1, cluster_x_2, cluster_y_2, cluster_z_2, count_1, charge_count_1, loc_eta_1, loc_phi_1, localDir0_1, localDir1_1, localDir2_1, lengthDir0_1, lengthDir1_1, lengthDir2_1, glob_eta_1, glob_phi_1, eta_angle_1, phi_angle_1, count_2, charge_count_2, loc_eta_2, loc_phi_2, localDir0_2, localDir1_2, localDir2_2, lengthDir0_2, lengthDir1_2, lengthDir2_2, glob_eta_2, glob_phi_2, eta_angle_2, phi_angle_2",
      "Feature names for the Embedding model"}

Definition at line 74 of file SiGNNTrackFinderTool.h.

                                          {
      this, "EmbeddingFeatureNames",
      "r, phi, z, cluster_x_1, cluster_y_1, cluster_z_1, cluster_x_2, cluster_y_2, cluster_z_2, count_1, charge_count_1, loc_eta_1, loc_phi_1, localDir0_1, localDir1_1, localDir2_1, lengthDir0_1, lengthDir1_1, lengthDir2_1, glob_eta_1, glob_phi_1, eta_angle_1, phi_angle_1, count_2, charge_count_2, loc_eta_2, loc_phi_2, localDir0_2, localDir1_2, localDir2_2, lengthDir0_2, lengthDir1_2, lengthDir2_2, glob_eta_2, glob_phi_2, eta_angle_2, phi_angle_2",
      "Feature names for the Embedding model"};

m_embeddingFeatureNamesVec

std::vector<std::string> InDet::SiGNNTrackFinderTool::m_embeddingFeatureNamesVec

private

Definition at line 117 of file SiGNNTrackFinderTool.h.

m_embeddingFeatureScales

StringProperty InDet::SiGNNTrackFinderTool::m_embeddingFeatureScales

protected

Initial value:

{
      this, "EmbeddingFeatureScales",
      "1000, 3.14, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14",
      "Feature scales for the Embedding model"}

Definition at line 78 of file SiGNNTrackFinderTool.h.

                                           {
      this, "EmbeddingFeatureScales",
      "1000, 3.14, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14",
      "Feature scales for the Embedding model"};

m_embeddingFeatureScalesVec

std::vector<float> InDet::SiGNNTrackFinderTool::m_embeddingFeatureScalesVec

private

Definition at line 118 of file SiGNNTrackFinderTool.h.

m_embedSessionTool

ToolHandle< AthOnnx::IOnnxRuntimeInferenceTool > InDet::SiGNNTrackFinderTool::m_embedSessionTool

private

Initial value:

{
      this, "Embedding", "AthOnnx::OnnxRuntimeInferenceTool"
    }

Definition at line 105 of file SiGNNTrackFinderTool.h.

                                                                      {
      this, "Embedding", "AthOnnx::OnnxRuntimeInferenceTool"
    };

m_filterCut

FloatProperty InDet::SiGNNTrackFinderTool::m_filterCut {this, "filterCut", 0.05}

protected

Definition at line 69 of file SiGNNTrackFinderTool.h.

{this, "filterCut", 0.05};

m_filterFeatureNames

StringProperty InDet::SiGNNTrackFinderTool::m_filterFeatureNames

protected

Initial value:

{
      this, "FilterFeatureNames",
      "r, phi, z, cluster_x_1, cluster_y_1, cluster_z_1, cluster_x_2, cluster_y_2, cluster_z_2, count_1, charge_count_1, loc_eta_1, loc_phi_1, localDir0_1, localDir1_1, localDir2_1, lengthDir0_1, lengthDir1_1, lengthDir2_1, glob_eta_1, glob_phi_1, eta_angle_1, phi_angle_1, count_2, charge_count_2, loc_eta_2, loc_phi_2, localDir0_2, localDir1_2, localDir2_2, lengthDir0_2, lengthDir1_2, lengthDir2_2, glob_eta_2, glob_phi_2, eta_angle_2, phi_angle_2",
      "Feature names for the Filtering model"}

Definition at line 83 of file SiGNNTrackFinderTool.h.

                                       {
      this, "FilterFeatureNames",
      "r, phi, z, cluster_x_1, cluster_y_1, cluster_z_1, cluster_x_2, cluster_y_2, cluster_z_2, count_1, charge_count_1, loc_eta_1, loc_phi_1, localDir0_1, localDir1_1, localDir2_1, lengthDir0_1, lengthDir1_1, lengthDir2_1, glob_eta_1, glob_phi_1, eta_angle_1, phi_angle_1, count_2, charge_count_2, loc_eta_2, loc_phi_2, localDir0_2, localDir1_2, localDir2_2, lengthDir0_2, lengthDir1_2, lengthDir2_2, glob_eta_2, glob_phi_2, eta_angle_2, phi_angle_2",
      "Feature names for the Filtering model"};

m_filterFeatureNamesVec

std::vector<std::string> InDet::SiGNNTrackFinderTool::m_filterFeatureNamesVec

private

Definition at line 119 of file SiGNNTrackFinderTool.h.

m_filterFeatureScales

StringProperty InDet::SiGNNTrackFinderTool::m_filterFeatureScales

protected

Initial value:

{
      this, "FilterFeatureScales",
      "1000, 3.14, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14",
      "Feature scales for the Filtering model"}

Definition at line 87 of file SiGNNTrackFinderTool.h.

                                        {
      this, "FilterFeatureScales",
      "1000, 3.14, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14, 1, 1, 3.14, 3.14, 1, 1, 1, 1, 1, 1, 3.14, 3.14, 3.14, 3.14",
      "Feature scales for the Filtering model"};

m_filterFeatureScalesVec

std::vector<float> InDet::SiGNNTrackFinderTool::m_filterFeatureScalesVec

private

Definition at line 120 of file SiGNNTrackFinderTool.h.

m_filterSessionTool

ToolHandle< AthOnnx::IOnnxRuntimeInferenceTool > InDet::SiGNNTrackFinderTool::m_filterSessionTool

private

Initial value:

{
      this, "Filtering", "AthOnnx::OnnxRuntimeInferenceTool"
    }

Definition at line 108 of file SiGNNTrackFinderTool.h.

                                                                       {
      this, "Filtering", "AthOnnx::OnnxRuntimeInferenceTool"
    };

m_gnnFeatureNames

StringProperty InDet::SiGNNTrackFinderTool::m_gnnFeatureNames

protected

Initial value:

{
      this, "GNNFeatureNames",
      "r, phi, z, eta, cluster_r_1, cluster_phi_1, cluster_z_1, cluster_eta_1, cluster_r_2, cluster_phi_2, cluster_z_2, cluster_eta_2",
      "Feature names for the GNN model"}

Definition at line 92 of file SiGNNTrackFinderTool.h.

                                    {
      this, "GNNFeatureNames",
      "r, phi, z, eta, cluster_r_1, cluster_phi_1, cluster_z_1, cluster_eta_1, cluster_r_2, cluster_phi_2, cluster_z_2, cluster_eta_2",
      "Feature names for the GNN model"};

m_gnnFeatureNamesVec

std::vector<std::string> InDet::SiGNNTrackFinderTool::m_gnnFeatureNamesVec

private

Definition at line 121 of file SiGNNTrackFinderTool.h.

m_gnnFeatureScales

StringProperty InDet::SiGNNTrackFinderTool::m_gnnFeatureScales

protected

Initial value:

{
      this, "GNNFeatureScales",
      "1000.0, 3.14159265359, 1000.0, 1.0, 1000.0, 3.14159265359, 1000.0, 1.0, 1000.0, 3.14159265359, 1000.0, 1.0",
      "Feature scales for the GNN model"}

Definition at line 96 of file SiGNNTrackFinderTool.h.

                                     {
      this, "GNNFeatureScales",
      "1000.0, 3.14159265359, 1000.0, 1.0, 1000.0, 3.14159265359, 1000.0, 1.0, 1000.0, 3.14159265359, 1000.0, 1.0",
      "Feature scales for the GNN model"};

m_gnnFeatureScalesVec

std::vector<float> InDet::SiGNNTrackFinderTool::m_gnnFeatureScalesVec

private

Definition at line 122 of file SiGNNTrackFinderTool.h.

m_gnnSessionTool

ToolHandle< AthOnnx::IOnnxRuntimeInferenceTool > InDet::SiGNNTrackFinderTool::m_gnnSessionTool

private

Initial value:

{
      this, "GNN", "AthOnnx::OnnxRuntimeInferenceTool"
    }

Definition at line 111 of file SiGNNTrackFinderTool.h.

                                                                    {
      this, "GNN", "AthOnnx::OnnxRuntimeInferenceTool"
    };

m_inputMLModuleDir

StringProperty InDet::SiGNNTrackFinderTool::m_inputMLModuleDir {this, "inputMLModelDir", ""}

protected

Definition at line 65 of file SiGNNTrackFinderTool.h.

{this, "inputMLModelDir", ""};

m_knnVal

UnsignedIntegerProperty InDet::SiGNNTrackFinderTool::m_knnVal {this, "knnVal", 1000}

protected

Definition at line 68 of file SiGNNTrackFinderTool.h.

{this, "knnVal", 1000};

m_rVal

FloatProperty InDet::SiGNNTrackFinderTool::m_rVal {this, "rVal", 0.12}

protected

Definition at line 67 of file SiGNNTrackFinderTool.h.

{this, "rVal", 0.12};

m_spacepointFeatureTool

ToolHandle<ISpacepointFeatureTool> InDet::SiGNNTrackFinderTool::m_spacepointFeatureTool

private

Initial value:

{
      this, "SpacepointFeatureTool", "InDet::SpacepointFeatureTool"}

Definition at line 114 of file SiGNNTrackFinderTool.h.

                                                              {
      this, "SpacepointFeatureTool", "InDet::SpacepointFeatureTool"};

m_walkMax

FloatProperty InDet::SiGNNTrackFinderTool::m_walkMax {this, "walkMax", 0.6}

protected

Definition at line 72 of file SiGNNTrackFinderTool.h.

{this, "walkMax", 0.6};

m_walkMin

FloatProperty InDet::SiGNNTrackFinderTool::m_walkMin {this, "walkMin", 0.1}

protected

Definition at line 71 of file SiGNNTrackFinderTool.h.

{this, "walkMin", 0.1};

---

The documentation for this class was generated from the following files:

• SiGNNTrackFinderTool.h
• SiGNNTrackFinderTool.cxx