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 Class Reference

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

Inheritance diagram for TrainVisitor:
Inheritance graph
[legend]

Public Types

template<template< typename TreeMetricType, typename TreeStatType, typename TreeMatType > class TreeType>
using RSTypeT = RSType< TreeType >
 Alias template necessary for visual c++ compiler. More...
 

Public Member Functions

 TrainVisitor (arma::mat &&referenceSet, const size_t leafSize)
 Construct the TrainVisitor object with the given reference set, leafSize. More...
 
template<template< typename TreeMetricType, typename TreeStatType, typename TreeMatType > class TreeType>
void operator() (RSTypeT< TreeType > *rs) const
 Default Train on the given RSType instance. More...
 
void operator() (RSTypeT< tree::KDTree > *rs) const
 Train on the given RSType specialized for KDTrees. More...
 
void operator() (RSTypeT< tree::BallTree > *rs) const
 Train on the given RSType specialized for BallTrees. More...
 
void operator() (RSTypeT< tree::Octree > *rs) const
 Train specialized for octrees. More...
 

Detailed Description

TrainVisitor sets the reference set to a new reference set on the given RSType.

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 130 of file rs_model.hpp.

Member Typedef Documentation

using RSTypeT = RSType<TreeType>

Alias template necessary for visual c++ compiler.

Definition at line 146 of file rs_model.hpp.

Constructor & Destructor Documentation

TrainVisitor ( arma::mat &&  referenceSet,
const size_t  leafSize 
)

Construct the TrainVisitor object with the given reference set, leafSize.

Member Function Documentation

void operator() ( RSTypeT< TreeType > *  rs) const

Default Train on the given RSType instance.

void operator() ( RSTypeT< tree::KDTree > *  rs) const

Train on the given RSType specialized for KDTrees.

void operator() ( RSTypeT< tree::BallTree > *  rs) const

Train on the given RSType specialized for BallTrees.

void operator() ( RSTypeT< tree::Octree > *  rs) const

Train specialized for octrees.


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