ActsPlugins::TorchEdgeClassifier::Config Struct Reference

#include <ActsPlugins/Gnn/TorchEdgeClassifier.hpp>

Collaboration diagram for ActsPlugins::TorchEdgeClassifier::Config:

Public Attributes
float	cut = 0.5
int	deviceID = 0
std::string	modelPath
int	nChunks = 1
std::vector< int >	selectedFeatures = {}
bool	undirected = false
bool	useEdgeFeatures = false

Member Data Documentation

cut

float ActsPlugins::TorchEdgeClassifier::Config::cut = 0.5

deviceID

int ActsPlugins::TorchEdgeClassifier::Config::deviceID = 0

modelPath

std::string ActsPlugins::TorchEdgeClassifier::Config::modelPath

nChunks

int ActsPlugins::TorchEdgeClassifier::Config::nChunks = 1

selectedFeatures

std::vector<int> ActsPlugins::TorchEdgeClassifier::Config::selectedFeatures = {}

undirected

bool ActsPlugins::TorchEdgeClassifier::Config::undirected = false

useEdgeFeatures

bool ActsPlugins::TorchEdgeClassifier::Config::useEdgeFeatures = false