ActsPlugins::TorchEdgeClassifier::Config Struct Reference

Configuration struct for Torch edge classifier. More...

#include <ActsPlugins/Gnn/TorchEdgeClassifier.hpp>

Collaboration diagram for ActsPlugins::TorchEdgeClassifier::Config:

Public Attributes
float	cut = 0.5
	Classification score threshold for edge filtering.
Device	device = Device::Cuda()
	Device to run inference on.
std::string	modelPath
	Path to the PyTorch model file.
int	nChunks = 1
	Number of chunks to process.
std::vector< int >	selectedFeatures = {}
	Selected feature indices for input.
bool	undirected = false
	Whether to treat graph as undirected.
bool	useEdgeFeatures = false
	Whether to use edge features.

Detailed Description

Configuration struct for Torch edge classifier.