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

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

  size_t End() const { return dimensions; }

  size_t Next() const { return dimensions; }

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

 private:
  size_t dimensions;
};

} // namespace tree
} // namespace mlpack

#endif