mlpack  master
Classes | Static Public Member Functions | List of all members
mlpack::tree::MinimalSplitsNumberSweep< SplitPolicy > Class Template Reference

The MinimalSplitsNumberSweep class finds a partition along which we can split a node according to the number of required splits of the node. More...

Classes

struct  SweepCost
 A struct that provides the type of the sweep cost. More...
 

Static Public Member Functions

template<typename TreeType >
static size_t SweepLeafNode (const size_t axis, const TreeType *node, typename TreeType::ElemType &axisCut)
 Find a suitable partition of a leaf node along the provided axis. More...
 
template<typename TreeType >
static size_t SweepNonLeafNode (const size_t axis, const TreeType *node, typename TreeType::ElemType &axisCut)
 Find a suitable partition of a non-leaf node along the provided axis. More...
 

Detailed Description

template<typename SplitPolicy>
class mlpack::tree::MinimalSplitsNumberSweep< SplitPolicy >

The MinimalSplitsNumberSweep class finds a partition along which we can split a node according to the number of required splits of the node.

The class finds a partition along a given axis. Moreover, the class evaluates the cost of each split. The cost is proportional to the number of required splits and the difference of sizes of resulting nodes. If the resulting nodes are overflowed the maximum cost is returned.

Template Parameters
SplitPolicyThe class that provides rules for inserting children of a node that is being split into two new subtrees.

Definition at line 31 of file minimal_splits_number_sweep.hpp.

Member Function Documentation

template<typename SplitPolicy >
template<typename TreeType >
static size_t mlpack::tree::MinimalSplitsNumberSweep< SplitPolicy >::SweepLeafNode ( const size_t  axis,
const TreeType *  node,
typename TreeType::ElemType &  axisCut 
)
static

Find a suitable partition of a leaf node along the provided axis.

The method returns the cost of the split.

Parameters
axisThe axis along which we are finding a partition.
nodeThe node that is being split.
axisCutThe coordinate at which the node may be split.
template<typename SplitPolicy >
template<typename TreeType >
static size_t mlpack::tree::MinimalSplitsNumberSweep< SplitPolicy >::SweepNonLeafNode ( const size_t  axis,
const TreeType *  node,
typename TreeType::ElemType &  axisCut 
)
static

Find a suitable partition of a non-leaf node along the provided axis.

The method returns the cost of the split.

Parameters
axisThe axis along which we are finding a partition.
nodeThe node that is being split.
axisCutThe coordinate at which the node may be split.

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