TMVA_SOFIE_GNN_Application.C File Reference

#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"

Include dependency graph for TMVA_SOFIE_GNN_Application.C:

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

Function Documentation

check_mem()

double check_mem

(

std::string

s = ""

)

Definition at line 27 of file TMVA_SOFIE_GNN_Application.C.

GenerateData()

std::vector< GNN_Data > GenerateData	(	int	nevts,
		int	seed )

Definition at line 109 of file TMVA_SOFIE_GNN_Application.C.

main()

int main

(

)

Definition at line 234 of file TMVA_SOFIE_GNN_Application.C.

Print()

void Print	(	GNN_Data &	d,
		std::string	txt = "" )

Definition at line 59 of file TMVA_SOFIE_GNN_Application.C.

PrintTensor()

template<class T >

void PrintTensor

(

RTensor< T > &

t

)

Definition at line 40 of file TMVA_SOFIE_GNN_Application.C.

ReadData()

std::vector< GNN_Data > ReadData	(	std::string	treename,
		std::string	filename )

Definition at line 136 of file TMVA_SOFIE_GNN_Application.C.

TMVA_SOFIE_GNN_Application()

void TMVA_SOFIE_GNN_Application

(

bool

verbose = false

)

Definition at line 171 of file TMVA_SOFIE_GNN_Application.C.

Variable Documentation

EDGE_FEATURE_SIZE

const int EDGE_FEATURE_SIZE = 4

Definition at line 106 of file TMVA_SOFIE_GNN_Application.C.

GLOBAL_FEATURE_SIZE

const int GLOBAL_FEATURE_SIZE = 1

Definition at line 107 of file TMVA_SOFIE_GNN_Application.C.

NODE_FEATURE_SIZE

const int NODE_FEATURE_SIZE = 4

Definition at line 105 of file TMVA_SOFIE_GNN_Application.C.

num_max_edges

const int num_max_edges = 30

Definition at line 104 of file TMVA_SOFIE_GNN_Application.C.

num_max_nodes

const int num_max_nodes = 10

Definition at line 103 of file TMVA_SOFIE_GNN_Application.C.