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 GRT_API LinearRegression : public Regressifier
{
    public:
    LinearRegression(bool useScaling=false);

    LinearRegression(const LinearRegression &rhs);
    
    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 save( std::fstream &file ) const;
    
    virtual bool load( std::fstream &file );
    
    UINT getMaxNumIterations() const;
    
    bool setMaxNumIterations(const UINT maxNumIterations);

    static std::string getId();
    
    //Tell the compiler we are using the base class train method to stop hidden virtual function warnings
    using MLBase::save;
    using MLBase::load;
    
protected:
    bool loadLegacyModelFromFile( std::fstream &file );
    
    Float w0;
    VectorFloat w;

private:
    static RegisterRegressifierModule< LinearRegression > registerModule;
    static const std::string id;
};

GRT_END_NAMESPACE

#endif //GRT_LINEAR_REGRESSION_HEADER