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. More...

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.

◆ m_embeddingDim

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

protected

Definition at line 66 of file SiGNNTrackFinderTool.h.

◆ 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.

◆ 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.

◆ 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.

◆ m_filterCut

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

protected

Definition at line 69 of file SiGNNTrackFinderTool.h.

◆ 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.

◆ 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.

◆ 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.

◆ 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.

◆ 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.

◆ 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.

◆ m_inputMLModuleDir

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

protected

Definition at line 65 of file SiGNNTrackFinderTool.h.

◆ m_knnVal

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

protected

Definition at line 68 of file SiGNNTrackFinderTool.h.

◆ m_rVal

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

protected

Definition at line 67 of file SiGNNTrackFinderTool.h.

◆ m_spacepointFeatureTool

ToolHandle<ISpacepointFeatureTool> InDet::SiGNNTrackFinderTool::m_spacepointFeatureTool

private

Initial value:

{

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

Definition at line 114 of file SiGNNTrackFinderTool.h.

◆ m_walkMax

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

protected

Definition at line 72 of file SiGNNTrackFinderTool.h.

◆ m_walkMin

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

protected

Definition at line 71 of file SiGNNTrackFinderTool.h.

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

Public Member Functions

Protected Member Functions

Exa.TrkX pipeline configurations, which will not be changed after construction

Detailed Description

Constructor & Destructor Documentation

◆ SiGNNTrackFinderTool() [1/3]

◆ SiGNNTrackFinderTool() [2/3]

◆ SiGNNTrackFinderTool() [3/3]

Member Function Documentation

◆ dump() [1/2]

◆ dump() [2/2]

◆ dumpevent()

◆ getTracks()

◆ initialize()

◆ initTrainedModels()

◆ operator=()

Member Data Documentation

◆ m_ccCut

◆ m_embeddingDim

◆ m_embeddingFeatureNames

◆ m_embeddingFeatureNamesVec

◆ m_embeddingFeatureScales

◆ m_embeddingFeatureScalesVec

◆ m_embedSessionTool

◆ m_filterCut

◆ m_filterFeatureNames

◆ m_filterFeatureNamesVec

◆ m_filterFeatureScales

◆ m_filterFeatureScalesVec

◆ m_filterSessionTool

◆ m_gnnFeatureNames

◆ m_gnnFeatureNamesVec

◆ m_gnnFeatureScales

◆ m_gnnFeatureScalesVec

◆ m_gnnSessionTool

◆ m_inputMLModuleDir

◆ m_knnVal

◆ m_rVal

◆ m_spacepointFeatureTool

◆ m_walkMax

◆ m_walkMin