LinearRegression.h

Go to the documentation of this file.

#ifndef GRT_LINEAR_REGRESSION_HEADER
#define GRT_LINEAR_REGRESSION_HEADER

#include "../../CoreModules/Regressifier.h"

GRT_BEGIN_NAMESPACE

class LinearRegression : public Regressifier
{
public:
 LinearRegression(bool useScaling=false);
 
 virtual ~LinearRegression(void);
 
 LinearRegression &operator=(const LinearRegression &rhs);
 
 virtual bool deepCopyFrom(const Regressifier *regressifier);
 
 virtual bool train_(RegressionData &trainingData);
 
 virtual bool predict_(VectorFloat &inputVector);
 
 virtual bool saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::fstream &file );
 
 UINT getMaxNumIterations() const;
 
 bool setMaxNumIterations(const UINT maxNumIterations);
 
 //Tell the compiler we are using the base class train method to stop hidden virtual function warnings
 using MLBase::saveModelToFile;
 using MLBase::loadModelFromFile;

protected:
 bool loadLegacyModelFromFile( std::fstream &file );
 
 Float w0;
 VectorFloat w;
 static RegisterRegressifierModule< LinearRegression > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_LINEAR_REGRESSION_HEADER