AbstractMLModel Class Reference

#include <AbstractMLModel.h>

Inheritance diagram for AbstractMLModel:

Collaboration diagram for AbstractMLModel:

Public Member Functions
	AbstractMLModel (const std::string &model_metadata)
	AbstractMLModel (const std::string &model_metadata, const std::vector< std::vector< std::string >> &cat_feature_keys)
virtual MLModelType	getModelType () const =0
virtual std::string	getModelTypeString () const =0
virtual int64_t	getNumFeatures () const =0
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

Protected Attributes
std::string	model_metadata_
std::vector< std::vector < std::string > >	cat_feature_keys_

Detailed Description

Definition at line 34 of file AbstractMLModel.h.

Constructor & Destructor Documentation

AbstractMLModel::AbstractMLModel ( const std::string &  model_metadata )

inline

Definition at line 36 of file AbstractMLModel.h.

      : model_metadata_(default_metadata(model_metadata)) {}

AbstractMLModel::AbstractMLModel ( const std::string &  model_metadata, const std::vector< std::vector< std::string >> &  cat_feature_keys  )

inline

Definition at line 39 of file AbstractMLModel.h.

      : model_metadata_(default_metadata(model_metadata))
      , cat_feature_keys_(cat_feature_keys) {}

virtual AbstractMLModel::~AbstractMLModel ( )

virtualdefault

Member Function Documentation

const std::vector<std::vector<std::string> >& AbstractMLModel::getCatFeatureKeys ( ) const

inline

Definition at line 58 of file AbstractMLModel.h.

References cat_feature_keys_.

                                                                   {
    return cat_feature_keys_;
  }

MLModelMetadata AbstractMLModel::getModelMetadata ( ) const

inline

Definition at line 48 of file AbstractMLModel.h.

References getModelMetadataStr(), getModelType(), getModelTypeString(), getNumCatFeatures(), getNumFeatures(), and getNumLogicalFeatures().

                                           {
    return MLModelMetadata("",
                           getModelType(),
                           getModelTypeString(),
                           getNumLogicalFeatures(),
                           getNumFeatures(),
                           getNumCatFeatures(),
                           getNumLogicalFeatures() - getNumCatFeatures(),
                           getModelMetadataStr());
  }

Here is the call graph for this function:

const std::string& AbstractMLModel::getModelMetadataStr ( ) const

inline

Definition at line 47 of file AbstractMLModel.h.

References model_metadata_.

Referenced by getModelMetadata().

{ return model_metadata_; }

Here is the caller graph for this function:

virtual MLModelType AbstractMLModel::getModelType ( ) const

pure virtual

Implemented in PcaModel, AbstractTreeModel, and LinearRegressionModel.

Referenced by getModelMetadata().

Here is the caller graph for this function:

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

pure virtual

Implemented in PcaModel, AbstractTreeModel, and LinearRegressionModel.

Referenced by getModelMetadata().

Here is the caller graph for this function:

const int64_t AbstractMLModel::getNumCatFeatures ( ) const

inline

Definition at line 61 of file AbstractMLModel.h.

References cat_feature_keys_.

Referenced by getModelMetadata(), and getNumLogicalFeatures().

{ return cat_feature_keys_.size(); }

Here is the caller graph for this function:

virtual int64_t AbstractMLModel::getNumFeatures ( ) const

pure virtual

Implemented in PcaModel, AbstractTreeModel, and LinearRegressionModel.

Referenced by getModelMetadata(), and getNumLogicalFeatures().

Here is the caller graph for this function:

const int64_t AbstractMLModel::getNumLogicalFeatures ( ) const

inline

Definition at line 71 of file AbstractMLModel.h.

References getNumCatFeatures(), getNumFeatures(), and getNumOneHotFeatures().

Referenced by getModelMetadata().

                                              {
    return getNumFeatures() - getNumOneHotFeatures() + getNumCatFeatures();
  }

Here is the call graph for this function:

Here is the caller graph for this function:

const int64_t AbstractMLModel::getNumOneHotFeatures ( ) const

inline

Definition at line 63 of file AbstractMLModel.h.

References cat_feature_keys_.

Referenced by getNumLogicalFeatures().

                                             {
    int64_t num_one_hot_features{0};
    for (const auto& cat_feature_key : cat_feature_keys_) {
      num_one_hot_features += static_cast<int64_t>(cat_feature_key.size());
    }
    return num_one_hot_features;
  }

Here is the caller graph for this function:

Member Data Documentation

std::vector<std::vector<std::string> > AbstractMLModel::cat_feature_keys_

protected

Definition at line 77 of file AbstractMLModel.h.

Referenced by getCatFeatureKeys(), getNumCatFeatures(), and getNumOneHotFeatures().

std::string AbstractMLModel::model_metadata_

protected

Definition at line 76 of file AbstractMLModel.h.

Referenced by getModelMetadataStr().

---

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

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