Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::RModel_GNN Class Referencefinal

Definition at line 98 of file RModel_GNN.hxx.

Public Member Functions

 RModel_GNN ()=default
 Default constructor.
 
 RModel_GNN (const RModel_GNN &other)=delete
 
 RModel_GNN (GNN_Init &graph_input_struct)
 
 RModel_GNN (RModel_GNN &&other)
 
 ~RModel_GNN () final=default
 
void Generate () final
 
RModel_GNNoperator= (const RModel_GNN &other)=delete
 
RModel_GNNoperator= (RModel_GNN &&other)
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::RModel_GNNBase
 RModel_GNNBase ()=default
 Default constructor.
 
virtual ~RModel_GNNBase ()=default
 
- Public Member Functions inherited from TMVA::Experimental::SOFIE::RModel_Base
 RModel_Base ()=default
 Default constructor.
 
 RModel_Base (std::string function_name)
 
 RModel_Base (std::string name, std::string parsedtime)
 
void AddBlasRoutines (std::vector< std::string > routines)
 
void AddNeededCustomHeader (std::string filename)
 
void AddNeededStdLib (std::string libname)
 
void GenerateHeaderInfo (std::string &hgname)
 
std::string GetFilename ()
 
const std::string & GetName () const
 
void OutputGenerated (std::string filename="", bool append=false)
 
void PrintGenerated ()
 
std::string ReturnGenerated ()
 
void SetFilename (std::string filename)
 

Private Attributes

std::unique_ptr< RFunction_Aggregateedge_global_agg_block
 
std::unique_ptr< RFunction_Aggregateedge_node_agg_block
 
std::unique_ptr< RFunction_Updateedges_update_block
 
std::unique_ptr< RFunction_Updateglobals_update_block
 
std::unique_ptr< RFunction_Aggregatenode_global_agg_block
 
std::unique_ptr< RFunction_Updatenodes_update_block
 
std::size_t num_edge_features
 
std::size_t num_edges
 
std::size_t num_global_features
 
std::size_t num_node_features
 
std::size_t num_nodes
 

Additional Inherited Members

- Protected Attributes inherited from TMVA::Experimental::SOFIE::RModel_Base
const std::unordered_set< std::string > fAllowedStdLib = {"vector", "algorithm", "cmath"}
 
std::unordered_set< std::string > fCustomOpHeaders
 
std::string fFileName
 
std::string fGC
 
bool fIsGNN = false
 
bool fIsGNNComponent = false
 
std::string fName = "UnnamedModel"
 
std::unordered_set< std::string > fNeededBlasRoutines
 
std::unordered_set< std::string > fNeededStdLib = {"vector"}
 
std::string fParseTime
 
bool fUseSession = true
 
bool fUseWeightFile = true
 
WeightFileType fWeightFile = WeightFileType::Text
 

#include <TMVA/RModel_GNN.hxx>

Inheritance diagram for TMVA::Experimental::SOFIE::RModel_GNN:
[legend]

Constructor & Destructor Documentation

◆ RModel_GNN() [1/4]

TMVA::Experimental::SOFIE::RModel_GNN::RModel_GNN ( )
default

Default constructor.

Needed to allow serialization of ROOT objects. See https://root.cern/manual/io_custom_classes/#restrictions-on-types-root-io-can-handle

◆ RModel_GNN() [2/4]

TMVA::Experimental::SOFIE::RModel_GNN::RModel_GNN ( GNN_Init graph_input_struct)

Definition at line 49 of file RModel_GNN.cxx.

◆ RModel_GNN() [3/4]

TMVA::Experimental::SOFIE::RModel_GNN::RModel_GNN ( RModel_GNN &&  other)

Definition at line 13 of file RModel_GNN.cxx.

◆ RModel_GNN() [4/4]

TMVA::Experimental::SOFIE::RModel_GNN::RModel_GNN ( const RModel_GNN other)
delete

◆ ~RModel_GNN()

TMVA::Experimental::SOFIE::RModel_GNN::~RModel_GNN ( )
finaldefault

Member Function Documentation

◆ Generate()

void TMVA::Experimental::SOFIE::RModel_GNN::Generate ( )
finalvirtual

Implements TMVA::Experimental::SOFIE::RModel_GNNBase.

Definition at line 72 of file RModel_GNN.cxx.

◆ operator=() [1/2]

RModel_GNN & TMVA::Experimental::SOFIE::RModel_GNN::operator= ( const RModel_GNN other)
delete

◆ operator=() [2/2]

RModel_GNN & TMVA::Experimental::SOFIE::RModel_GNN::operator= ( RModel_GNN &&  other)

Definition at line 30 of file RModel_GNN.cxx.

Member Data Documentation

◆ edge_global_agg_block

std::unique_ptr<RFunction_Aggregate> TMVA::Experimental::SOFIE::RModel_GNN::edge_global_agg_block
private

Definition at line 108 of file RModel_GNN.hxx.

◆ edge_node_agg_block

std::unique_ptr<RFunction_Aggregate> TMVA::Experimental::SOFIE::RModel_GNN::edge_node_agg_block
private

Definition at line 107 of file RModel_GNN.hxx.

◆ edges_update_block

std::unique_ptr<RFunction_Update> TMVA::Experimental::SOFIE::RModel_GNN::edges_update_block
private

Definition at line 102 of file RModel_GNN.hxx.

◆ globals_update_block

std::unique_ptr<RFunction_Update> TMVA::Experimental::SOFIE::RModel_GNN::globals_update_block
private

Definition at line 104 of file RModel_GNN.hxx.

◆ node_global_agg_block

std::unique_ptr<RFunction_Aggregate> TMVA::Experimental::SOFIE::RModel_GNN::node_global_agg_block
private

Definition at line 109 of file RModel_GNN.hxx.

◆ nodes_update_block

std::unique_ptr<RFunction_Update> TMVA::Experimental::SOFIE::RModel_GNN::nodes_update_block
private

Definition at line 103 of file RModel_GNN.hxx.

◆ num_edge_features

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_edge_features
private

Definition at line 115 of file RModel_GNN.hxx.

◆ num_edges

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_edges
private

Definition at line 112 of file RModel_GNN.hxx.

◆ num_global_features

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_global_features
private

Definition at line 116 of file RModel_GNN.hxx.

◆ num_node_features

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_node_features
private

Definition at line 114 of file RModel_GNN.hxx.

◆ num_nodes

std::size_t TMVA::Experimental::SOFIE::RModel_GNN::num_nodes
private

Definition at line 111 of file RModel_GNN.hxx.

  • tmva/sofie/inc/TMVA/RModel_GNN.hxx
  • tmva/sofie/src/RModel_GNN.cxx