ActsPlugins::TorchMetricLearning::Config Struct Reference

#include <ActsPlugins/Gnn/TorchMetricLearning.hpp>

Collaboration diagram for ActsPlugins::TorchMetricLearning::Config:

Public Attributes
int	deviceID = 0
int	embeddingDim = 8
int	knnVal = 500
std::string	modelPath
float	phiScale = 3.141592654
float	rVal = 1.6
std::vector< int >	selectedFeatures = {}
bool	shuffleDirections = false

Member Data Documentation

deviceID

int ActsPlugins::TorchMetricLearning::Config::deviceID = 0

embeddingDim

int ActsPlugins::TorchMetricLearning::Config::embeddingDim = 8

knnVal

int ActsPlugins::TorchMetricLearning::Config::knnVal = 500

modelPath

std::string ActsPlugins::TorchMetricLearning::Config::modelPath

phiScale

float ActsPlugins::TorchMetricLearning::Config::phiScale = 3.141592654

rVal

float ActsPlugins::TorchMetricLearning::Config::rVal = 1.6

selectedFeatures

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

shuffleDirections

bool ActsPlugins::TorchMetricLearning::Config::shuffleDirections = false