hoeffding_categorical_split.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_HOEFFDING_TREES_HOEFFDING_CATEGORICAL_SPLIT_HPP
#define MLPACK_METHODS_HOEFFDING_TREES_HOEFFDING_CATEGORICAL_SPLIT_HPP
 
#include <mlpack/prereqs.hpp>
#include "categorical_split_info.hpp"
 
namespace mlpack {
namespace tree {
 
template<typename FitnessFunction>
class HoeffdingCategoricalSplit
{
 public:
  typedef CategoricalSplitInfo SplitInfo;
 
  HoeffdingCategoricalSplit(const size_t numCategories = 0,
                            const size_t numClasses = 0);
 
  HoeffdingCategoricalSplit(const size_t numCategories,
                            const size_t numClasses,
                            const HoeffdingCategoricalSplit& other);
 
  template<typename eT>
  void Train(eT value, const size_t label);
 
  void EvaluateFitnessFunction(double& bestFitness, double& secondBestFitness)
      const;
 
  size_t NumChildren() const { return sufficientStatistics.n_cols; }
 
  void Split(arma::Col<size_t>& childMajorities, SplitInfo& splitInfo);
 
  size_t MajorityClass() const;
  double MajorityProbability() const;
 
  template<typename Archive>
  void serialize(Archive& ar, const unsigned int /* version */)
  {
    ar & BOOST_SERIALIZATION_NVP(sufficientStatistics);
  }
 
 private:
  arma::Mat<size_t> sufficientStatistics;
};
 
} // namespace tree
} // namespace mlpack
 
// Include implementation.
#include "hoeffding_categorical_split_impl.hpp"
 
#endif