minimal_splits_number_sweep.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_TREE_RECTANGLE_TREE_MINIMAL_SPLITS_NUMBER_SWEEP_HPP
#define MLPACK_CORE_TREE_RECTANGLE_TREE_MINIMAL_SPLITS_NUMBER_SWEEP_HPP
 
namespace mlpack {
namespace tree {
 
template<typename SplitPolicy>
class MinimalSplitsNumberSweep
{
 public:
  template<typename>
  struct SweepCost
  {
    typedef size_t type;
  };
 
  template<typename TreeType>
  static size_t SweepNonLeafNode(
      const size_t axis,
      const TreeType* node,
      typename TreeType::ElemType& axisCut);
 
  template<typename TreeType>
  static size_t SweepLeafNode(
      const size_t axis,
      const TreeType* node,
      typename TreeType::ElemType& axisCut);
};
 
} // namespace tree
} // namespace mlpack
 
// Include implementation
#include "minimal_splits_number_sweep_impl.hpp"
 
#endif  //  MLPACK_CORE_TREE_RECTANGLE_TREE_MINIMAL_SPLITS_NUMBER_SWEEP_HPP