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;
 }
 
 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