ml_linear_model.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ML_LINEAR_MODEL_H
#define ML_LINEAR_MODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_global.h"
#include "ml_model.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// DEFINE NAMESPACE MLLIB
//=============================================================================================================
 
namespace MLLIB{
 
//=============================================================================================================
class MLSHARED_EXPORT MlLinearModel : public MlModel
{
public:
    //=========================================================================================================
    MlLinearModel(MlTaskType type = MlTaskType::Regression, double regularization = 1.0);
 
    //=========================================================================================================
    MlTensor predict(const MlTensor& input) const override;
    bool save(const QString& path) const override;
    bool load(const QString& path) override;
    QString modelType() const override;
    MlTaskType taskType() const override;
 
    //=========================================================================================================
    const Eigen::MatrixXf& weights() const;
 
    //=========================================================================================================
    const Eigen::VectorXf& bias() const;
 
private:
    MlTaskType      m_taskType;             
    double          m_regularization;       
    Eigen::MatrixXf m_weights;              
    Eigen::VectorXf m_bias;                 
    bool            m_trained = false;      
};
 
} // namespace MLLIB
 
#endif // ML_LINEAR_MODEL_H