random_dimension_select.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_DECISION_TREE_RANDOM_DIMENSION_SELECT_HPP
#define MLPACK_METHODS_DECISION_TREE_RANDOM_DIMENSION_SELECT_HPP

namespace mlpack {
namespace tree {

class RandomDimensionSelect
{
 public:
  RandomDimensionSelect(const size_t dimensions) : dimensions(dimensions) { }

  size_t Begin() const { return math::RandInt(dimensions); }

  size_t End() const { return dimensions; }

  size_t Next() const { return dimensions; }

 private:
  const size_t dimensions;
};

} // namespace tree
} // namespace mlpack

#endif