RBMQuantizer.h

Go to the documentation of this file.

#ifndef GRT_RBM_QUANTIZER_HEADER
#define GRT_RBM_QUANTIZER_HEADER

//Include the main GRT header to get access to the FeatureExtraction base class
#include "../../CoreModules/FeatureExtraction.h"
#include "../../CoreAlgorithms/BernoulliRBM/BernoulliRBM.h"
#include "../../DataStructures/TimeSeriesClassificationData.h"
#include "../../DataStructures/ClassificationDataStream.h"
#include "../../DataStructures/UnlabelledData.h"

GRT_BEGIN_NAMESPACE
    
class RBMQuantizer : public FeatureExtraction{
public:
    RBMQuantizer(const UINT numClusters=10);
 
    RBMQuantizer(const RBMQuantizer &rhs);
    
    virtual ~RBMQuantizer();
    
    RBMQuantizer& operator=(const RBMQuantizer &rhs);

    virtual bool deepCopyFrom(const FeatureExtraction *featureExtraction);
    
    virtual bool computeFeatures(const VectorFloat &inputVector);
    
    virtual bool reset();
    
    virtual bool clear();
    
    virtual bool saveModelToFile( std::string filename ) const;
    
    virtual bool loadModelFromFile( std::string filename );
    
    virtual bool saveModelToFile( std::fstream &file ) const;
    
    virtual bool loadModelFromFile( std::fstream &file );

    virtual bool train_(ClassificationData &trainingData);

    virtual bool train_(TimeSeriesClassificationData &trainingData);

    virtual bool train_(ClassificationDataStream &trainingData);

    virtual bool train_(UnlabelledData &trainingData);

    virtual bool train_(MatrixFloat &trainingData);

 UINT quantize(const Float inputValue);
 
 UINT quantize(const VectorFloat &inputVector);
    
    bool getQuantizerTrained() const;

    UINT getNumClusters() const;
    
 UINT getQuantizedValue() const;
 
 VectorFloat getQuantizationDistances() const;
 
 BernoulliRBM getBernoulliRBM() const;
    
    bool setNumClusters(const UINT numClusters);
    
    //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
    using MLBase::train;
    using MLBase::train_;
    using MLBase::predict;
    using MLBase::predict_;
    
protected:
    UINT numClusters;
    BernoulliRBM rbm;
    VectorFloat quantizationDistances;
    
    static RegisterFeatureExtractionModule< RBMQuantizer > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_RBM_QUANTIZER_HEADER