RadialBasisFunction.h

Go to the documentation of this file.

#ifndef GRT_RADIAL_BASIC_FUNCTION_HEADER
#define GRT_RADIAL_BASIC_FUNCTION_HEADER

#include "WeakClassifier.h"

GRT_BEGIN_NAMESPACE
    
class GRT_API RadialBasisFunction : public WeakClassifier{
public:
    RadialBasisFunction(UINT numSteps=100,Float positiveClassificationThreshold=0.9,Float minAlphaSearchRange=0.001,Float maxAlphaSearchRange=1.0);
    
    virtual ~RadialBasisFunction();
    
    RadialBasisFunction(const RadialBasisFunction &rhs);
    
    RadialBasisFunction& operator=(const RadialBasisFunction &rhs);
    
    virtual bool deepCopyFrom(const WeakClassifier *weakClassifer);
    
    virtual bool train(ClassificationData &trainingData, VectorFloat &weights);
    
    virtual Float predict(const VectorFloat &x);
    
    virtual bool saveModelToFile( std::fstream &file ) const;
    
    virtual bool loadModelFromFile( std::fstream &file );
    
    virtual void print() const;
    
    UINT getNumSteps() const;
    
    Float getPositiveClassificationThreshold() const;
    
    Float getAlpha() const;
    
    Float getMinAlphaSearchRange() const;
    
    Float getMaxAlphaSearchRange() const;
    
    VectorFloat getRBFCentre() const;
    
protected:
    Float rbf(const VectorFloat &a,const VectorFloat &b);
  
    UINT numSteps;
    Float positiveClassificationThreshold;
    Float alpha;
    Float gamma;
    Float minAlphaSearchRange;
    Float maxAlphaSearchRange;
    VectorFloat rbfCentre;
    
    static RegisterWeakClassifierModule< RadialBasisFunction > registerModule; 
};

GRT_END_NAMESPACE

#endif // GRT_DECISION_STUMP_HEADER