mlpack  2.2.5
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Public Types | Public Member Functions | List of all members
TrainVisitor< SortPolicy > Class Template Reference

TrainVisitor sets the reference set to a new reference set on the given NSType. More...

Inheritance diagram for TrainVisitor< SortPolicy >:
Inheritance graph
[legend]

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...
 

Detailed Description

template<typename SortPolicy>
class mlpack::neighbor::TrainVisitor< SortPolicy >

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.

Member Typedef Documentation

using NSTypeT = NSType<SortPolicy, TreeType>

Alias template necessary for visual c++ compiler.

Definition at line 183 of file ns_model.hpp.

Constructor & Destructor Documentation

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.

Member Function Documentation

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.


The documentation for this class was generated from the following files: