BAG.h

Go to the documentation of this file.

#ifndef GRT_BAG_HEADER
#define GRT_BAG_HEADER

#include "../../CoreModules/Classifier.h"

GRT_BEGIN_NAMESPACE

class BAG : public Classifier
{
public:
 BAG(bool useScaling=false);
 
 BAG(const BAG &rhs);
 
 virtual ~BAG(void);
 
 BAG &operator=(const BAG &rhs);
 
 virtual bool deepCopyFrom(const Classifier *classifier);
 
 virtual bool train_(ClassificationData &trainingData);
 
 virtual bool predict_(VectorFloat &inputVector);
 
 virtual bool reset();
 
 virtual bool clear();
 
 virtual bool saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::fstream &file );
 
 UINT getEnsembleSize() const;
 
 VectorFloat getEnsembleWeights() const;
 
 const Vector< Classifier* > getEnsemble() const;

 bool addClassifierToEnsemble(const Classifier &classifier,Float weight=1);
 
 bool clearEnsemble();
 
 bool setWeights(const VectorFloat &weights);
 
 //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
 using MLBase::saveModelToFile;
 using MLBase::loadModelFromFile;
 using MLBase::train;
 using MLBase::train_;
 using MLBase::predict;
 using MLBase::predict_;
 
protected:
 bool loadLegacyModelFromFile( std::fstream &file );
 
 VectorFloat weights;
 Vector< Classifier* > ensemble;
 
 static RegisterClassifierModule< BAG > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_BAG_HEADER