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 GRT_API ClassLabelFilter : public PostProcessing{
public:
 ClassLabelFilter(const UINT minimumCount = 1,const UINT bufferSize = 1);
 
 ClassLabelFilter(const ClassLabelFilter &rhs);
 
 virtual ~ClassLabelFilter();
 
 ClassLabelFilter& operator=(const ClassLabelFilter &rhs);
 
 virtual bool deepCopyFrom(const PostProcessing *postProcessing) override;
 
 virtual bool process(const VectorDouble &inputVector) override;
 
 virtual bool reset() override;
 
 virtual bool save( std::fstream &file ) const override;
 
 virtual bool load( std::fstream &file ) override;
 
 bool init(const UINT minimumCount,const UINT bufferSize);
 
 UINT filter(const UINT predictedClassLabel);
 
 UINT getFilteredClassLabel() const { return filteredClassLabel; }
 
 bool setMinimumCount(const UINT minimumCount);
 
 bool setBufferSize(const UINT bufferSize);

 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;
 
protected:
 UINT filteredClassLabel; 
 UINT minimumCount; 
 UINT bufferSize; 
 CircularBuffer< UINT > buffer; 
 
private:
 static const std::string id; 
 static RegisterPostProcessingModule< ClassLabelFilter > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_CLASS_LABEL_FILTER_HEADER