WeakClassifier.h

Go to the documentation of this file.

#ifndef GRT_WEAK_CLASSIFIER_HEADER
#define GRT_WEAK_CLASSIFIER_HEADER

#include "../../../Util/GRTCommon.h"
#include "../../../DataStructures/ClassificationData.h"

GRT_BEGIN_NAMESPACE
    
#define WEAK_CLASSIFIER_POSITIVE_CLASS_LABEL 1
#define WEAK_CLASSIFIER_NEGATIVE_CLASS_LABEL 2
    
class GRT_API WeakClassifier{
public:
    WeakClassifier();
    
    virtual ~WeakClassifier();
    
    WeakClassifier(const WeakClassifier &rhs){
        *this = rhs;
    }
    
    WeakClassifier& operator=(const WeakClassifier &rhs);
    
    bool copyBaseVariables(const WeakClassifier *weakClassifer);
    
    virtual bool deepCopyFrom(const WeakClassifier *weakClassifer){
        return false;
    }
    
    virtual bool train(ClassificationData &trainingData, VectorFloat &weights){
        return false;
    }
    
    virtual Float predict(const VectorFloat &x){
        return 0;
    }
    
    virtual bool saveModelToFile( std::fstream &file ) const{ return false; }
    
    virtual bool loadModelFromFile( std::fstream &file ){ return false; }
    
    virtual void print() const{}
    
    virtual Float getPositiveClassLabel() const{ return 1; }
    
    virtual Float getNegativeClassLabel() const{ return -1; }
    
    std::string getWeakClassifierType() const{
        return weakClassifierType;
    }
    
    bool getTrained() const{
        return trained;
    }
    
    UINT getNumInputDimensions() const{
        return numInputDimensions;
    }

    bool getTrainingLoggingEnabled() const{
        return trainingLog.getLoggingEnabled();
    }

    bool setTrainingLoggingEnabled(const bool enabled){
        return trainingLog.setLoggingEnabled(enabled);
    }
    
    typedef std::map< std::string, WeakClassifier*(*)() > StringWeakClassifierMap;
    
    static WeakClassifier* createInstanceFromString( std::string const &weakClassifierType );
    
    WeakClassifier* createNewInstance() const;
    
protected:
    std::string weakClassifierType;  
    bool trained;               
    UINT numInputDimensions;    
    TrainingLog trainingLog;
    ErrorLog errorLog;
    WarningLog warningLog;
    
    static StringWeakClassifierMap *getMap() {
        if( !stringWeakClassifierMap ){ stringWeakClassifierMap = new StringWeakClassifierMap; }
        return stringWeakClassifierMap;
    }
    
private:
    static StringWeakClassifierMap *stringWeakClassifierMap;
    static UINT numWeakClassifierInstances;
};
    
//These two functions/classes are used to register any new WeakClassification Module with the WeakClassifier base class
template< typename T >  WeakClassifier *newWeakClassificationModuleInstance() { return new T; }

template< typename T >
class RegisterWeakClassifierModule : WeakClassifier {
public:
    RegisterWeakClassifierModule( std::string const &newWeakClassificationModuleName ) {
        getMap()->insert( std::pair< std::string, WeakClassifier*(*)() >(newWeakClassificationModuleName, &newWeakClassificationModuleInstance< T > ) );
    }
};

GRT_END_NAMESPACE

#endif //GRT_WEAK_CLASSIFIER_HEADER