all_dimension_select.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_DECISION_TREE_ALL_DIMENSION_SELECT_HPP
#define MLPACK_METHODS_DECISION_TREE_ALL_DIMENSION_SELECT_HPP

namespace mlpack {
namespace tree {

class AllDimensionSelect
{
 public:
  AllDimensionSelect(const size_t dimensions) : i(0), dimensions(dimensions) { }

  size_t Begin()
  {
    i = 0;
    return 0;
  }

  size_t End() const { return dimensions; }

  size_t Next() { return ++i; }

 private:
  size_t i;
  const size_t dimensions;
};

} // namespace tree
} // namespace mlpack

#endif