LinearRegressionModel Class Reference

#include <MLModel.h>

Inheritance diagram for LinearRegressionModel:

Collaboration diagram for LinearRegressionModel:

Public Member Functions
	LinearRegressionModel (const std::vector< double > &coefs, const std::string &model_metadata)
	LinearRegressionModel (const std::vector< double > &coefs, const std::string &model_metadata, const std::vector< std::vector< std::string >> &cat_feature_keys)
virtual MLModelType	getModelType () const override
virtual std::string	getModelTypeString () const override
virtual int64_t	getNumFeatures () const override
const std::vector< double > &	getCoefs () const
Public Member Functions inherited from AbstractMLModel
	AbstractMLModel (const std::string &model_metadata)
	AbstractMLModel (const std::string &model_metadata, const std::vector< std::vector< std::string >> &cat_feature_keys)
virtual	~AbstractMLModel ()=default
const std::string &	getModelMetadataStr () const
MLModelMetadata	getModelMetadata () const
const std::vector< std::vector < std::string > > &	getCatFeatureKeys () const
const int64_t	getNumCatFeatures () const
const int64_t	getNumOneHotFeatures () const
const int64_t	getNumLogicalFeatures () const

Private Attributes
std::vector< double >	coefs_

Additional Inherited Members
Protected Attributes inherited from AbstractMLModel
std::string	model_metadata_
std::vector< std::vector < std::string > >	cat_feature_keys_

Detailed Description

Definition at line 126 of file MLModel.h.

Constructor & Destructor Documentation

LinearRegressionModel::LinearRegressionModel ( const std::vector< double > &  coefs, const std::string &  model_metadata  )

inline

Definition at line 128 of file MLModel.h.

      : AbstractMLModel(model_metadata), coefs_(coefs) {}

LinearRegressionModel::LinearRegressionModel ( const std::vector< double > &  coefs, const std::string &  model_metadata, const std::vector< std::vector< std::string >> &  cat_feature_keys  )

inline

Definition at line 132 of file MLModel.h.

      : AbstractMLModel(model_metadata, cat_feature_keys), coefs_(coefs) {}

Member Function Documentation

const std::vector<double>& LinearRegressionModel::getCoefs ( ) const

inline

Definition at line 145 of file MLModel.h.

References coefs_.

Referenced by ShowModelFeatureDetailsCommand::extractExtraMetadata().

{ return coefs_; }

Here is the caller graph for this function:

virtual MLModelType LinearRegressionModel::getModelType ( ) const

inlineoverridevirtual

Implements AbstractMLModel.

Definition at line 137 of file MLModel.h.

References LINEAR_REG.

{ return MLModelType::LINEAR_REG; }

virtual std::string LinearRegressionModel::getModelTypeString ( ) const

inlineoverridevirtual

Implements AbstractMLModel.

Definition at line 139 of file MLModel.h.

{ return "Linear Regression"; }

virtual int64_t LinearRegressionModel::getNumFeatures ( ) const

inlineoverridevirtual

Implements AbstractMLModel.

Definition at line 141 of file MLModel.h.

References coefs_.

                                                  {
    return static_cast<int64_t>(coefs_.size()) - 1;
  }

Member Data Documentation

std::vector<double> LinearRegressionModel::coefs_

private

Definition at line 148 of file MLModel.h.

Referenced by getCoefs(), and getNumFeatures().

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/TableFunctions/SystemFunctions/os/ML/MLModel.h