adaboost.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ADABOOST_ADABOOST_HPP
#define MLPACK_METHODS_ADABOOST_ADABOOST_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/methods/perceptron/perceptron.hpp>
#include <mlpack/methods/decision_stump/decision_stump.hpp>

namespace mlpack {
namespace adaboost {

template<typename WeakLearnerType = mlpack::perceptron::Perceptron<>,
         typename MatType = arma::mat>
class AdaBoost
{
 public:
  AdaBoost(const MatType& data,
           const arma::Row<size_t>& labels,
           const size_t numClasses,
           const WeakLearnerType& other,
           const size_t iterations = 100,
           const double tolerance = 1e-6);

  AdaBoost(const double tolerance = 1e-6);

  // Return the value of ztProduct.
  double ZtProduct() { return ztProduct; }

  double Tolerance() const { return tolerance; }
  double& Tolerance() { return tolerance; }

  size_t NumClasses() const { return numClasses; }

  size_t WeakLearners() const { return alpha.size(); }

  double Alpha(const size_t i) const { return alpha[i]; }
  double& Alpha(const size_t i) { return alpha[i]; }

  const WeakLearnerType& WeakLearner(const size_t i) const { return wl[i]; }
  WeakLearnerType& WeakLearner(const size_t i) { return wl[i]; }

  void Train(const MatType& data,
             const arma::Row<size_t>& labels,
             const size_t numClasses,
             const WeakLearnerType& learner,
             const size_t iterations = 100,
             const double tolerance = 1e-6);

  void Classify(const MatType& test, arma::Row<size_t>& predictedLabels);

  template<typename Archive>
  void serialize(Archive& ar, const unsigned int /* version */);

 private:
  size_t numClasses;
  // The tolerance for change in rt and when to stop.
  double tolerance;

  std::vector<WeakLearnerType> wl;
  std::vector<double> alpha;

  double ztProduct;
}; // class AdaBoost

} // namespace adaboost
} // namespace mlpack

#include "adaboost_impl.hpp"

#endif