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 GRT_API 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 save( std::fstream &file ) const;
    
    virtual bool load( std::fstream &file );
    
    virtual bool recomputeNullRejectionThresholds();
    
    UINT getNumClusters() const;
    
    Vector< MinDistModel > getModels() const;
    
    virtual bool setNullRejectionCoeff(Float nullRejectionCoeff);
    
    bool setNumClusters(UINT numClusters);
    
    static std::string getId();

    //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
    using MLBase::save;
    using MLBase::load;
    using MLBase::train_;
    using MLBase::predict_;
    
protected:
    bool loadLegacyModelFromFile( std::fstream &file );
    
    UINT numClusters;
    Vector< MinDistModel > models;            //A buffer to hold all the models
    
private:
    static const std::string id;
    static RegisterClassifierModule< MinDist > registerModule;
};

} //End of namespace GRT

#endif //GRT_MINDIST_HEADER