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 GRT_API 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 save( std::fstream &file ) const;
    
    virtual bool load( 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);
    
    static std::string getId();

    //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
    using MLBase::save;
    using MLBase::load;
    using MLBase::train_;
    using MLBase::predict_;
    
protected:
    bool loadLegacyModelFromFile( std::fstream &file );
    
    VectorFloat weights;
    Vector< Classifier* > ensemble;
    
private:
    static RegisterClassifierModule< BAG > registerModule;
    static const std::string id;
};

GRT_END_NAMESPACE

#endif //GRT_BAG_HEADER