MultidimensionalRegression.h

Go to the documentation of this file.

#ifndef GRT_MULTIDIMENSIONAL_REGRESSION_HEADER
#define GRT_MULTIDIMENSIONAL_REGRESSION_HEADER

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

GRT_BEGIN_NAMESPACE

class GRT_API MultidimensionalRegression : public Regressifier
{
public:
 MultidimensionalRegression(const Regressifier &regressifier = LinearRegression(),bool useScaling=false);

    MultidimensionalRegression(const MultidimensionalRegression &rhs);
    
 virtual ~MultidimensionalRegression(void);
    
 MultidimensionalRegression &operator=(const MultidimensionalRegression &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 );
    
    bool getIsRegressionModuleSet() const;
    
    Regressifier *getRegressifier() const;
    
    bool setRegressionModule( const Regressifier &regressifier );

    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 deepCopyRegressionModules( Vector< Regressifier* > &newModules ) const;
    bool deleteAll();
 bool deleteRegressionModules();
    bool loadLegacyModelFromFile( std::fstream &file );
 
    Regressifier *regressifier;
 Vector< Regressifier* > regressionModules;

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

GRT_END_NAMESPACE

#endif //GRT_MULTIDIMENSIONAL_REGRESSION_HEADER