LogisticRegression.h

Go to the documentation of this file.

#ifndef GRT_LOGISTIC_REGRESSION_HEADER
#define GRT_LOGISTIC_REGRESSION_HEADER

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

GRT_BEGIN_NAMESPACE

class LogisticRegression : public Regressifier
{
public:
 LogisticRegression(const bool useScaling=true);
    
 virtual ~LogisticRegression(void);
    
 LogisticRegression &operator=(const LogisticRegression &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(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:
 inline Float sigmoid(const Float x) const;
    bool loadLegacyModelFromFile( std::fstream &file );
 
    Float w0; 
    VectorFloat w; 
    static RegisterRegressifierModule< LogisticRegression > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_LOGISTIC_REGRESSION_HEADER