TrainVisitor Class Reference

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

Inheritance diagram for TrainVisitor:

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

Member Typedef Documentation

◆ RSTypeT

using RSTypeT = RSType<TreeType>

Alias template necessary for visual c++ compiler.

Definition at line 141 of file rs_model.hpp.

Constructor & Destructor Documentation

◆ TrainVisitor()

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

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

Member Function Documentation

◆ operator()() [1/4]

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

Default Train on the given RSType instance.

◆ operator()() [2/4]

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

Train on the given RSType specialized for KDTrees.

◆ operator()() [3/4]

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

Train on the given RSType specialized for BallTrees.

◆ operator()() [4/4]

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

Train specialized for octrees.


The documentation for this class was generated from the following file:
  • /home/jenkins-mlpack/mlpack.org/_src/mlpack-3.2.1/src/mlpack/methods/range_search/rs_model.hpp