FiniteStateMachine.h

Go to the documentation of this file.

#ifndef GRT_FINITE_STATE_MACHINE_HEADER
#define GRT_FINITE_STATE_MACHINE_HEADER

#include "../../CoreModules/Classifier.h"
#include "FSMParticleFilter.h"
#include "../../ClusteringModules/KMeans/KMeans.h"

GRT_BEGIN_NAMESPACE

class GRT_API FiniteStateMachine : public Classifier
{
public:
    
    FiniteStateMachine(const UINT numParticles = 200,const UINT numClustersPerState = 20,const Float stateTransitionSmoothingCoeff = 0.0,const Float measurementNoise = 10.0);
    
    FiniteStateMachine(const FiniteStateMachine &rhs);
    
    virtual ~FiniteStateMachine(void);
    
    FiniteStateMachine &operator=(const FiniteStateMachine &rhs);
    
    virtual bool deepCopyFrom(const Classifier *classifier);
    
    virtual bool train_( ClassificationData &trainingData );
    
    virtual bool train_( TimeSeriesClassificationData &trainingData );
    
    virtual bool train_( TimeSeriesClassificationDataStream &data );
    
    virtual bool predict_(VectorFloat &inputVector);
    
    virtual bool reset();
    
    virtual bool clear();
    
    virtual bool print() const;
    
    virtual bool save( std::fstream &file ) const;
    
    virtual bool load( std::fstream &file );
    
    
    bool setNumParticles(const UINT numParticles);
    bool setNumClustersPerState(const UINT numClustersPerState);
    bool setStateTransitionSmoothingCoeff(const Float stateTransitionSmoothingCoeff);
    bool setMeasurementNoise(const Float measurementNoise);
    
    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 recomputePT();
    bool recomputePE();
    bool initParticles();
    
    UINT numParticles;
    UINT numClustersPerState;
    Float stateTransitionSmoothingCoeff;
    Float measurementNoise;
    FSMParticleFilter particles;
    MatrixFloat stateTransitions;
    Vector< MatrixFloat > stateEmissions;
    Vector< Vector< IndexedDouble > > pt;   
    Vector< Vector< VectorFloat > > pe;    
    
private:
    static RegisterClassifierModule< FiniteStateMachine > registerModule;
    static const std::string id;
};

GRT_END_NAMESPACE

#endif