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 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 saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::fstream &file );
 
 
 bool setNumParticles(const UINT numParticles);
 bool setNumClustersPerState(const UINT numClustersPerState);
 bool setStateTransitionSmoothingCoeff(const Float stateTransitionSmoothingCoeff);
 bool setMeasurementNoise(const Float measurementNoise);
 
 //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
 using MLBase::saveModelToFile;
 using MLBase::loadModelFromFile;
 using MLBase::train;
 using MLBase::train_;
 using MLBase::predict;
 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; 
 
 static RegisterClassifierModule< FiniteStateMachine > registerModule;
};

GRT_END_NAMESPACE

#endif