categorical_split_info.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_HOEFFDING_TREES_CATEGORICAL_SPLIT_INFO_HPP
#define MLPACK_METHODS_HOEFFDING_TREES_CATEGORICAL_SPLIT_INFO_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace tree {
 
class CategoricalSplitInfo
{
 public:
  CategoricalSplitInfo(const size_t /* categories */) { }
 
  template<typename eT>
  static size_t CalculateDirection(const eT& value)
  {
    // We have a child for each categorical value, and value should be in the
    // range [0, categories).
    return size_t(value);
  }
 
  template<typename Archive>
  void serialize(Archive& /* ar */, const unsigned int /* version */) { }
};
 
} // namespace tree
} // namespace mlpack
 
#endif