mlpack
2.2.5
|
TrainVisitor sets the reference set to a new reference set on the given NSType. More...
Public Types | |
template<template< typename TreeMetricType, typename TreeStatType, typename TreeMatType > class TreeType> | |
using | NSTypeT = NSType< SortPolicy, TreeType > |
Alias template necessary for visual c++ compiler. More... | |
Public Member Functions | |
TrainVisitor (arma::mat &&referenceSet, const size_t leafSize, const double tau, const double rho) | |
Construct the TrainVisitor object with the given reference set, leafSize for BinarySpaceTrees, and tau and rho for spill trees. More... | |
template<template< typename TreeMetricType, typename TreeStatType, typename TreeMatType > class TreeType> | |
void | operator() (NSTypeT< TreeType > *ns) const |
Default Train on the given NSType instance. More... | |
void | operator() (NSTypeT< tree::KDTree > *ns) const |
Train on the given NSType specialized for KDTrees. More... | |
void | operator() (NSTypeT< tree::BallTree > *ns) const |
Train on the given NSType specialized for BallTrees. More... | |
void | operator() (SpillKNN *ns) const |
Train specialized for SPTrees. More... | |
void | operator() (NSTypeT< tree::Octree > *ns) const |
Train specialized for octrees. More... | |
TrainVisitor sets the reference set to a new reference set on the given NSType.
We use template specialization to differentiate those tree types that accept leafSize as a parameter. In these cases, a reference tree with proper leafSize is built from the referenceSet.
Definition at line 35 of file neighbor_search.hpp.
Alias template necessary for visual c++ compiler.
Definition at line 183 of file ns_model.hpp.
TrainVisitor | ( | arma::mat && | referenceSet, |
const size_t | leafSize, | ||
const double | tau, | ||
const double | rho | ||
) |
Construct the TrainVisitor object with the given reference set, leafSize for BinarySpaceTrees, and tau and rho for spill trees.
void operator() | ( | NSTypeT< TreeType > * | ns | ) | const |
Default Train on the given NSType instance.
void operator() | ( | NSTypeT< tree::KDTree > * | ns | ) | const |
Train on the given NSType specialized for KDTrees.
void operator() | ( | NSTypeT< tree::BallTree > * | ns | ) | const |
Train on the given NSType specialized for BallTrees.
void operator() | ( | SpillKNN * | ns | ) | const |
Train specialized for SPTrees.
void operator() | ( | NSTypeT< tree::Octree > * | ns | ) | const |
Train specialized for octrees.