MLModelType.h File Reference

#include "Logger/Logger.h"
#include "Shared/StringTransform.h"
#include "Shared/misc.h"
#include <string>

Include dependency graph for MLModelType.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Enumerations
enum	MLModelType {   LINEAR_REG, DECISION_TREE_REG, GBT_REG, RANDOM_FOREST_REG,   PCA }

Functions
std::string	get_ml_model_type_str (const MLModelType model_type)
MLModelType	get_ml_model_type_from_str (const std::string &model_type_str)
bool	is_regression_model (const MLModelType model_type)

Enumeration Type Documentation

enum MLModelType

Enumerator
LINEAR_REG
DECISION_TREE_REG
GBT_REG
RANDOM_FOREST_REG
PCA

Definition at line 25 of file MLModelType.h.

{ LINEAR_REG, DECISION_TREE_REG, GBT_REG, RANDOM_FOREST_REG, PCA };

Function Documentation

MLModelType get_ml_model_type_from_str ( const std::string &  model_type_str )

inline

Definition at line 52 of file MLModelType.h.

References DECISION_TREE_REG, GBT_REG, LINEAR_REG, PCA, RANDOM_FOREST_REG, and to_upper().

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

                                                                               {
  const auto upper_model_type_str = to_upper(model_type_str);
  if (upper_model_type_str == "LINEAR_REG") {
    return MLModelType::LINEAR_REG;
  } else if (upper_model_type_str == "DECISION_TREE_REG") {
    return MLModelType::DECISION_TREE_REG;
  } else if (upper_model_type_str == "GBT_REG") {
    return MLModelType::GBT_REG;
  } else if (upper_model_type_str == "RANDOM_FOREST_REG") {
    return MLModelType::RANDOM_FOREST_REG;
  } else if (upper_model_type_str == "PCA") {
    return MLModelType::PCA;
  } else {
    throw std::invalid_argument("Unknown model type: " + upper_model_type_str);
  }
}

Here is the call graph for this function:

Here is the caller graph for this function:

std::string get_ml_model_type_str ( const MLModelType  model_type )

inline

Definition at line 27 of file MLModelType.h.

References CHECK, DECISION_TREE_REG, GBT_REG, LINEAR_REG, PCA, and RANDOM_FOREST_REG.

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

                                                                     {
  switch (model_type) {
    case MLModelType::LINEAR_REG: {
      return "LINEAR_REG";
    }
    case MLModelType::DECISION_TREE_REG: {
      return "DECISION_TREE_REG";
    }
    case MLModelType::GBT_REG: {
      return "GBT_REG";
    }
    case MLModelType::RANDOM_FOREST_REG: {
      return "RANDOM_FOREST_REG";
    }
    case MLModelType::PCA: {
      return "PCA";
    }
    default: {
      CHECK(false) << "Unknown model type.";
      // Satisfy compiler
      return "LINEAR_REG";
    }
  }
}

Here is the caller graph for this function:

bool is_regression_model ( const MLModelType  model_type )

inline

Definition at line 69 of file MLModelType.h.

References DECISION_TREE_REG, GBT_REG, shared::is_any(), LINEAR_REG, and RANDOM_FOREST_REG.

Referenced by Parser::CreateModelStmt::build_model_query(), and EvaluateModelCommand::execute().

                                                              {
  return shared::is_any<MLModelType::LINEAR_REG,
                        MLModelType::DECISION_TREE_REG,
                        MLModelType::GBT_REG,
                        MLModelType::RANDOM_FOREST_REG>(model_type);
}

Here is the call graph for this function:

Here is the caller graph for this function: