ClassLabelFilter.h

Go to the documentation of this file.

#ifndef GRT_CLASS_LABEL_FILTER_HEADER
#define GRT_CLASS_LABEL_FILTER_HEADER

#include "../Util/GRTCommon.h"
#include "../CoreModules/PostProcessing.h"

GRT_BEGIN_NAMESPACE

class ClassLabelFilter : public PostProcessing{
public:
 ClassLabelFilter(UINT minimumCount = 1,UINT bufferSize = 1);
 
 ClassLabelFilter(const ClassLabelFilter &rhs);
 
 virtual ~ClassLabelFilter();
 
 ClassLabelFilter& operator=(const ClassLabelFilter &rhs);
 
 virtual bool deepCopyFrom(const PostProcessing *postProcessing);
 
 virtual bool process(const VectorDouble &inputVector);
 
 virtual bool reset();
 
 virtual bool saveModelToFile( std::string filename ) const;
 
 virtual bool saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::string filename );
 
 virtual bool loadModelFromFile( std::fstream &file );
 
 bool init(UINT minimumCount,UINT bufferSize); 
 
 UINT filter(UINT predictedClassLabel);
 
 UINT getFilteredClassLabel(){ return filteredClassLabel; }
 
 bool setMinimumCount(UINT minimumCount);
 
 bool setBufferSize(UINT bufferSize);
 
protected:
 UINT filteredClassLabel; 
 UINT minimumCount; 
 UINT bufferSize; 
 CircularBuffer< UINT > buffer; 
 
 static RegisterPostProcessingModule< ClassLabelFilter > registerModule;
};
 
GRT_END_NAMESPACE

#endif //GRT_CLASS_LABEL_FILTER_HEADER