MLModelMap Class Reference

#include <MLModel.h>

Public Member Functions
void	addModel (const std::string &model_name, std::shared_ptr< AbstractMLModel > model)
bool	modelExists (const std::string &model_name) const
std::shared_ptr< AbstractMLModel >	getModel (const std::string &model_name) const
void	deleteModel (const std::string &model_name)
std::vector< std::string >	getModelNames () const
std::vector< MLModelMetadata >	getModelMetadata () const
MLModelMetadata	getModelMetadata (const std::string &model_name) const

Private Attributes
std::map< std::string, std::shared_ptr < AbstractMLModel > >	model_map_
std::shared_mutex	model_map_mutex_

Detailed Description

Definition at line 36 of file MLModel.h.

Member Function Documentation

void MLModelMap::addModel ( const std::string &  model_name, std::shared_ptr< AbstractMLModel >  model  )

inline

Definition at line 38 of file MLModel.h.

References model_map_, model_map_mutex_, and to_upper().

Referenced by linear_reg_fit_impl(), and pca_fit_impl().

                                                                                   {
    const auto upper_model_name = to_upper(model_name);
    std::lock_guard<std::shared_mutex> model_map_write_lock(model_map_mutex_);
    model_map_[upper_model_name] = model;
  }

Here is the call graph for this function:

Here is the caller graph for this function:

void MLModelMap::deleteModel ( const std::string &  model_name )

inline

Definition at line 62 of file MLModel.h.

References model_map_, model_map_mutex_, and to_upper().

Referenced by Parser::DropModelStmt::execute().

                                                {
    const auto upper_model_name = to_upper(model_name);
    std::lock_guard<std::shared_mutex> model_map_write_lock(model_map_mutex_);
    auto const model_it = model_map_.find(upper_model_name);
    if (model_it == model_map_.end()) {
      std::ostringstream error_oss;
      error_oss << "Cannot erase model " << upper_model_name
                << ". No model by that name was found.";
      throw std::runtime_error(error_oss.str());
    }
    model_map_.erase(model_it);
  }

Here is the call graph for this function:

Here is the caller graph for this function:

std::shared_ptr<AbstractMLModel> MLModelMap::getModel ( const std::string &  model_name ) const

inline

Definition at line 51 of file MLModel.h.

References model_map_, model_map_mutex_, and to_upper().

Referenced by CodeGenerator::codegen(), ShowModelFeatureDetailsCommand::execute(), get_decision_trees__cpu_1(), linear_reg_coefs__cpu_1(), ml_reg_predict__cpu_template(), r2_score__cpu_template(), and random_forest_reg_var_importance__cpu_1().

                                                                             {
    const auto upper_model_name = to_upper(model_name);
    std::shared_lock<std::shared_mutex> model_map_read_lock(model_map_mutex_);
    auto model_map_itr = model_map_.find(upper_model_name);
    if (model_map_itr != model_map_.end()) {
      return model_map_itr->second;
    }
    const std::string error_str = "Model '" + upper_model_name + "' does not exist.";
    throw std::runtime_error(error_str);
  }

Here is the call graph for this function:

Here is the caller graph for this function:

std::vector<MLModelMetadata> MLModelMap::getModelMetadata ( ) const

inline

Definition at line 84 of file MLModel.h.

References model_map_, and model_map_mutex_.

Referenced by ShowModelDetailsCommand::execute(), EvaluateModelCommand::execute(), and foreign_storage::InternalMLModelMetadataDataWrapper::initializeObjectsForTable().

                                                      {
    std::shared_lock<std::shared_mutex> model_map_read_lock(model_map_mutex_);
    std::vector<MLModelMetadata> model_metadata;
    for (auto const& model : model_map_) {
      model_metadata.emplace_back(MLModelMetadata(
          model.first,
          model.second->getModelType(),
          model.second->getModelTypeString(),
          model.second->getNumLogicalFeatures(),
          model.second->getNumFeatures(),
          model.second->getNumCatFeatures(),
          model.second->getNumLogicalFeatures() - model.second->getNumCatFeatures(),
          model.second->getModelMetadataStr()));
    }
    return model_metadata;
  }

Here is the caller graph for this function:

MLModelMetadata MLModelMap::getModelMetadata ( const std::string &  model_name ) const

inline

Definition at line 101 of file MLModel.h.

References model_map_, model_map_mutex_, and to_upper().

                                                                      {
    const auto upper_model_name = to_upper(model_name);
    std::shared_lock<std::shared_mutex> model_map_read_lock(model_map_mutex_);
    auto model_map_itr = model_map_.find(upper_model_name);
    if (model_map_itr != model_map_.end()) {
      return MLModelMetadata(model_map_itr->first,
                             model_map_itr->second->getModelType(),
                             model_map_itr->second->getModelTypeString(),
                             model_map_itr->second->getNumLogicalFeatures(),
                             model_map_itr->second->getNumFeatures(),
                             model_map_itr->second->getNumCatFeatures(),
                             model_map_itr->second->getNumLogicalFeatures() -
                                 model_map_itr->second->getNumCatFeatures(),
                             model_map_itr->second->getModelMetadataStr());
    }
    const std::string error_str = "Model '" + upper_model_name + "' does not exist.";
    throw std::runtime_error(error_str);
  }

Here is the call graph for this function:

std::vector<std::string> MLModelMap::getModelNames ( ) const

inline

Definition at line 75 of file MLModel.h.

References model_map_, and model_map_mutex_.

Referenced by ShowModelsCommand::execute(), and ShowModelDetailsCommand::getFilteredModelNames().

                                             {
    std::shared_lock<std::shared_mutex> model_map_read_lock(model_map_mutex_);
    std::vector<std::string> model_names;
    model_names.reserve(model_map_.size());
    for (auto const& model : model_map_) {
      model_names.emplace_back(model.first);
    }
    return model_names;
  }

Here is the caller graph for this function:

bool MLModelMap::modelExists ( const std::string &  model_name ) const

inline

Definition at line 44 of file MLModel.h.

References model_map_, model_map_mutex_, and to_upper().

Referenced by Parser::CreateModelStmt::check_model_exists().

                                                      {
    const auto upper_model_name = to_upper(model_name);
    std::shared_lock<std::shared_mutex> model_map_read_lock(model_map_mutex_);
    auto model_map_itr = model_map_.find(upper_model_name);
    return model_map_itr != model_map_.end();
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

std::map<std::string, std::shared_ptr<AbstractMLModel> > MLModelMap::model_map_

private

Definition at line 121 of file MLModel.h.

Referenced by addModel(), deleteModel(), getModel(), getModelMetadata(), getModelNames(), and modelExists().

std::shared_mutex MLModelMap::model_map_mutex_

mutableprivate

Definition at line 122 of file MLModel.h.

Referenced by addModel(), deleteModel(), getModel(), getModelMetadata(), getModelNames(), and modelExists().

---

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