RegressionData.h

Go to the documentation of this file.

#ifndef GRT_REGRESSION_DATA_HEADER
#define GRT_REGRESSION_DATA_HEADER

#include "../Util/GRTTypedefs.h"
 #include "VectorFloat.h"
#include "RegressionSample.h"

GRT_BEGIN_NAMESPACE

class RegressionData{
public:
 
 RegressionData(const UINT numInputDimensions=0,const UINT numTargetDimensions=0,const std::string datasetName="NOT_SET",const std::string infoText="");
 
 RegressionData(const RegressionData &rhs);
 
 ~RegressionData();

 RegressionData& operator=(const RegressionData &rhs);
 
 inline RegressionSample& operator[](const UINT &i){
 return data[i];
 }

 const inline RegressionSample& operator[](const UINT &i) const{
 return data[i];
 }
 
 void clear();

 bool setInputAndTargetDimensions(const UINT numInputDimensions,const UINT numTargetDimensions);
 
 bool setDatasetName(const std::string &datasetName);
 
 bool setInfoText(const std::string &infoText);
 
 bool addSample(const VectorFloat &inputVector,const VectorFloat &targetVector);
 
 bool removeLastSample();
 
 bool reserve(const UINT N);
 
 bool setExternalRanges(const Vector< MinMax > &externalInputRanges, const Vector< MinMax > &externalTargetRanges, const bool useExternalRanges);
 
 bool enableExternalRangeScaling(const bool useExternalRanges);
 
 bool scale(const Float minTarget,const Float maxTarget);
 
 bool scale(const Vector< MinMax > &inputVectorRanges,const Vector< MinMax > &targetVectorRanges,const Float minTarget,const Float maxTarget);
 
 bool save(const std::string &filename) const;
 
 bool load(const std::string &filename);
 
 bool saveDatasetToFile(const std::string &filename) const;
 
 bool loadDatasetFromFile(const std::string &filename);
 
 bool saveDatasetToCSVFile(const std::string &filename) const;
 
 bool loadDatasetFromCSVFile(const std::string &filename,const UINT numInputDimensions,const UINT numTargetDimensions);
 
 bool printStats() const;
 
 bool merge(const RegressionData &regressionData);
 
 RegressionData partition(const UINT trainingSizePercentage);
 
 bool spiltDataIntoKFolds(const UINT K);
 
 RegressionData getTrainingFoldData(const UINT foldIndex) const;
 
 RegressionData getTestFoldData(const UINT foldIndex) const;

 UINT removeDuplicateSamples();
 
 std::string getDatasetName()const { return datasetName; }
 
 std::string getInfoText() const { return infoText; }

 std::string getStatsAsString() const;
 
 UINT inline getNumInputDimensions() const{ return numInputDimensions; }
 
 UINT inline getNumTargetDimensions() const{ return numTargetDimensions; }
 
 UINT inline getNumSamples() const{ return totalNumSamples; }
 
 Vector<MinMax> getInputRanges() const;
 
 Vector<MinMax> getTargetRanges() const;
 
 Vector< RegressionSample > getData() const{ return data; }

private:
 std::string datasetName; 
 std::string infoText; 
 UINT numInputDimensions; 
 UINT numTargetDimensions; 
 UINT totalNumSamples; 
 UINT kFoldValue; 
 bool crossValidationSetup; 
 bool useExternalRanges; 
 Vector< MinMax > externalInputRanges; 
 Vector< MinMax > externalTargetRanges; 
 Vector< RegressionSample > data; 
 Vector< Vector< UINT > > crossValidationIndexs; 
 
 DebugLog debugLog; 
 ErrorLog errorLog; 
 WarningLog warningLog; 
};

GRT_END_NAMESPACE

#endif //GRT_REGRESSION_DATA_HEADER