MinDist.h

Go to the documentation of this file.

#ifndef GRT_MINDIST_HEADER
#define GRT_MINDIST_HEADER

#include "MinDistModel.h"
#include "../../CoreModules/Classifier.h"

GRT_BEGIN_NAMESPACE

class MinDist : public Classifier
{
public:
 MinDist(bool useScaling=false,bool useNullRejection=false,Float nullRejectionCoeff=10.0,UINT numClusters=10);
 
 MinDist(const MinDist &rhs);
 
 virtual ~MinDist(void);
 
 MinDist &operator=(const MinDist &rhs);
 
 virtual bool deepCopyFrom(const Classifier *classifier);
 
 virtual bool train_(ClassificationData &trainingData);
 
 virtual bool predict_(VectorFloat &inputVector);
 
 virtual bool clear();
 
 virtual bool saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::fstream &file );
 
 virtual bool recomputeNullRejectionThresholds();
 
 UINT getNumClusters() const;
 
 Vector< MinDistModel > getModels() const;
 
 virtual bool setNullRejectionCoeff(Float nullRejectionCoeff);
 
 bool setNumClusters(UINT numClusters);

protected:
 bool loadLegacyModelFromFile( std::fstream &file );
 
 UINT numClusters; 
 Vector< MinDistModel > models; //A buffer to hold all the models
 
private:
 static RegisterClassifierModule< MinDist > registerModule;
};

} //End of namespace GRT

#endif //GRT_MINDIST_HEADER