multiple_random_dimension_select.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_DECISION_TREE_MULTIPLE_RANDOM_DIMENSION_SPLIT_HPP
#define MLPACK_METHODS_DECISION_TREE_MULTIPLE_RANDOM_DIMENSION_SPLIT_HPP

namespace mlpack {
namespace tree {

template<size_t NumDimensions = 3>
class MultipleRandomDimensionSelect
{
 public:
  MultipleRandomDimensionSelect(const size_t dimensions)
  {
    for (size_t i = 0; i < NumDimensions; ++i)
    {
      // Generate random different numbers.
      bool unique = false;
      size_t value;
      while (!unique)
      {
        value = math::RandInt(dimensions);

        // Check if we already have the value.
        unique = true;
        for (size_t j = 0; j < i; ++j)
        {
          if (values[j] == value)
          {
            unique = false;
            break;
          }
        }
      }

      values[i] = value;
    }

    values[NumDimensions] = std::numeric_limits<size_t>::max();
  }

  size_t Begin()
  {
    i = 0;
    return values[0];
  }

  size_t End() const { return size_t(-1); }

  size_t Next()
  {
    return values[++i];
  }

 private:
  size_t values[NumDimensions + 1];
  size_t i;
};

} // namespace tree
} // namespace mlpack

#endif