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 GRT_API LogisticRegression : public Regressifier
{
public:
 LogisticRegression(const bool useScaling=true,const Float learningRate=0.01,const Float minChange=1.0e-5,const UINT batchSize=1,const UINT maxNumEpochs=500,const UINT minNumEpochs=1);

 LogisticRegression(const LogisticRegression &rhs);
 
 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 save( std::fstream &file ) const;
 
 virtual bool load( std::fstream &file );
 
 UINT getMaxNumIterations() const;
 
 bool setMaxNumIterations(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:
 inline Float sigmoid(const Float x) const;
 bool loadLegacyModelFromFile( std::fstream &file );
 
 Float w0; 
 VectorFloat w; 

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

GRT_END_NAMESPACE

#endif //GRT_LOGISTIC_REGRESSION_HEADER