KMeansFeatures.h

Go to the documentation of this file.

#ifndef GRT_KMEANS_FEATURES_HEADER
#define GRT_KMEANS_FEATURES_HEADER

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

GRT_BEGIN_NAMESPACE
 
class GRT_API KMeansFeatures : public FeatureExtraction{
public:
 KMeansFeatures(const Vector< UINT > numClustersPerLayer = Vector< UINT >(1,100),const Float alpha = 0.2,const bool useScaling = true);
 
 KMeansFeatures(const KMeansFeatures &rhs);
 
 virtual ~KMeansFeatures();
 
 KMeansFeatures& operator=(const KMeansFeatures &rhs);

 virtual bool deepCopyFrom(const FeatureExtraction *featureExtraction);
 
 virtual bool computeFeatures(const VectorFloat &inputVector);
 
 virtual bool reset();
 
 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);
 
 bool computeFeatures(VectorFloat &inputVector,VectorFloat &outputVector);
 
 bool init( const Vector< UINT > numClustersPerLayer );
 
 bool projectDataThroughLayer( const VectorFloat &input, VectorFloat &output, const UINT layer );
 
 UINT getNumLayers() const;
 UINT getLayerSize(const UINT layerIndex) const;
 Vector< MatrixFloat > getClusters() const;
 
 //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:
 Float alpha;
 Vector< UINT > numClustersPerLayer;
 Vector< MinMax > ranges;
 Vector< MatrixFloat > clusters;
 
 static RegisterFeatureExtractionModule< KMeansFeatures > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_KMEANS_FEATURES_HEADER