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