ActsPlugins::TensorRTEdgeClassifier::Config Struct Reference

Configuration struct for TensorRT edge classifier. More...

#include <ActsPlugins/Gnn/TensorRTEdgeClassifier.hpp>

Collaboration diagram for ActsPlugins::TensorRTEdgeClassifier::Config:

Public Attributes
float	cut = 0.5
	Classification score threshold for edge filtering.
std::string	modelPath
	Path to the TensorRT model file.
std::size_t	numExecutionContexts = 1
	Number of parallel execution contexts for inference.
std::vector< int >	selectedFeatures
	List of feature indices to use for edge classification.

Detailed Description

Configuration struct for TensorRT edge classifier.