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() : i(0), dimensions(0) { }

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

  size_t End() const { return dimensions; }

  size_t Next() { return ++i; }

  size_t Dimensions() const { return dimensions; }
  size_t& Dimensions() { return dimensions; }

 private:
  size_t i;
  size_t dimensions;
};

} // namespace tree
} // namespace mlpack

#endif