#include "encoder.hxx"#include "core.hxx"#include "decoder.hxx"#include "output_transform.hxx"#include "TMVA/SOFIE_common.hxx"#include "TRandom3.h"#include "TH1.h"#include "TCanvas.h"#include "TFile.h"#include "TTree.h"#include "TSystem.h"#include "ROOT/RDataFrame.hxx"| Classes | |
| struct | SOFIE_GNN | 
| Functions | |
| double | check_mem (std::string s="") | 
| std::vector< GNN_Data > | GenerateData (int nevts, int seed) | 
| int | main () | 
| void | Print (GNN_Data &d, std::string txt="") | 
| template<class T > | |
| void | PrintTensor (RTensor< T > &t) | 
| std::vector< GNN_Data > | ReadData (std::string treename, std::string filename) | 
| void | TMVA_SOFIE_GNN_Application (bool verbose=false) | 
| Variables | |
| const int | EDGE_FEATURE_SIZE = 4 | 
| const int | GLOBAL_FEATURE_SIZE = 1 | 
| const int | NODE_FEATURE_SIZE = 4 | 
| const int | num_max_edges = 30 | 
| const int | num_max_nodes = 10 | 
Definition in file TMVA_SOFIE_GNN_Application.C.
| double check_mem | ( | std::string | s = "" | ) | 
Definition at line 31 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 113 of file TMVA_SOFIE_GNN_Application.C.
| int main | ( | ) | 
Definition at line 238 of file TMVA_SOFIE_GNN_Application.C.
| void Print | ( | GNN_Data & | d, | 
| std::string | txt = "" ) | 
Definition at line 63 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 44 of file TMVA_SOFIE_GNN_Application.C.
| std::vector< GNN_Data > ReadData | ( | std::string | treename, | 
| std::string | filename ) | 
Definition at line 140 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 175 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 110 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 111 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 109 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 108 of file TMVA_SOFIE_GNN_Application.C.
Definition at line 107 of file TMVA_SOFIE_GNN_Application.C.