Bagging.h

Go to the documentation of this file.

#ifndef OPENANN_BAGGING_H_
#define OPENANN_BAGGING_H_

#include <OpenANN/EnsembleLearner.h>
#include <list>

namespace OpenANN
{

class Bagging : public EnsembleLearner
{
  std::list<Learner*> models;
  Optimizer* optimizer;
  double bagSize;
  int F;
public:
  Bagging(double bagSize);
  virtual EnsembleLearner& addLearner(Learner& learner);
  virtual EnsembleLearner& setOptimizer(Optimizer& optimizer);
  virtual EnsembleLearner& train(DataSet& dataSet);
  virtual Eigen::MatrixXd operator()(Eigen::MatrixXd& X);
  virtual Eigen::VectorXd operator()(Eigen::VectorXd& x);
};

} // namespace OpenANN

#endif // OPENANN_BAGGING_H_