OmniSciDB  72c90bc290
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
MLTableFunctionsCommon.h File Reference
+ Include dependency graph for MLTableFunctionsCommon.h:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Enumerations

enum  MLFramework { MLFramework::DEFAULT, MLFramework::ONEDAL, MLFramework::MLPACK, MLFramework::INVALID }
 
enum  KMeansInitStrategy {
  KMeansInitStrategy::DEFAULT, KMeansInitStrategy::DETERMINISTIC, KMeansInitStrategy::RANDOM, KMeansInitStrategy::PLUS_PLUS,
  KMeansInitStrategy::INVALID
}
 
enum  VarImportanceMetric {
  VarImportanceMetric::DEFAULT, VarImportanceMetric::NONE, VarImportanceMetric::MDI, VarImportanceMetric::MDA,
  VarImportanceMetric::MDA_SCALED, VarImportanceMetric::INVALID
}
 

Functions

MLFramework get_ml_framework (const std::string &ml_framework_str)
 
KMeansInitStrategy get_kmeans_init_type (const std::string &init_type_str)
 
VarImportanceMetric get_var_importance_metric (const std::string &var_importance_metric_str)
 

Enumeration Type Documentation

enum KMeansInitStrategy
strong
Enumerator
DEFAULT 
DETERMINISTIC 
RANDOM 
PLUS_PLUS 
INVALID 

Definition at line 38 of file MLTableFunctionsCommon.h.

enum MLFramework
strong
Enumerator
DEFAULT 
ONEDAL 
MLPACK 
INVALID 

Definition at line 23 of file MLTableFunctionsCommon.h.

Function Documentation

KMeansInitStrategy get_kmeans_init_type ( const std::string &  init_type_str)
inline

Definition at line 40 of file MLTableFunctionsCommon.h.

References DEFAULT, DETERMINISTIC, INVALID, PLUS_PLUS, RANDOM, and to_upper().

Referenced by kmeans__cpu_template().

40  {
41  const auto upper_init_type_str = to_upper(init_type_str);
42  const static std::map<std::string, KMeansInitStrategy> kmeans_init_type_map = {
43  {"DEFAULT", KMeansInitStrategy::DEFAULT},
44  {"DETERMINISTIC", KMeansInitStrategy::DETERMINISTIC},
45  {"RANDOM", KMeansInitStrategy::RANDOM},
46  {"PLUS_PLUS", KMeansInitStrategy::PLUS_PLUS}};
47  const auto itr = kmeans_init_type_map.find(upper_init_type_str);
48  if (itr == kmeans_init_type_map.end()) {
50  }
51  return itr->second;
52 }
std::string to_upper(const std::string &str)

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

MLFramework get_ml_framework ( const std::string &  ml_framework_str)
inline

Definition at line 25 of file MLTableFunctionsCommon.h.

References DEFAULT, INVALID, MLPACK, ONEDAL, and to_upper().

Referenced by dbscan__cpu_template(), decision_tree_reg_impl(), gbt_reg_fit_impl(), kmeans__cpu_template(), linear_reg_fit_impl(), ml_reg_predict_impl(), pca_fit_impl(), and random_forest_reg_fit_impl().

25  {
26  const auto upper_ml_framework_str = to_upper(ml_framework_str);
27  const static std::map<std::string, MLFramework> ml_framework_map = {
28  {"DEFAULT", MLFramework::DEFAULT},
29  {"ONEDAL", MLFramework::ONEDAL},
30  {"MLPACK", MLFramework::MLPACK}};
31  const auto itr = ml_framework_map.find(upper_ml_framework_str);
32  if (itr == ml_framework_map.end()) {
33  return MLFramework::INVALID;
34  }
35  return itr->second;
36 }
std::string to_upper(const std::string &str)

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

VarImportanceMetric get_var_importance_metric ( const std::string &  var_importance_metric_str)
inline

Definition at line 56 of file MLTableFunctionsCommon.h.

References DEFAULT, INVALID, MDA, MDA_SCALED, MDI, NONE, and to_upper().

Referenced by random_forest_reg_fit_impl().

57  {
58  const auto upper_var_importance_metric_str = to_upper(var_importance_metric_str);
59  const static std::map<std::string, VarImportanceMetric> var_importance_metric_map = {
60  {"DEFAULT", VarImportanceMetric::DEFAULT},
61  {"NONE", VarImportanceMetric::NONE},
62  {"MDI", VarImportanceMetric::MDI},
63  {"MDA", VarImportanceMetric::MDA},
64  {"MDA_SCALED", VarImportanceMetric::MDA_SCALED}};
65  const auto itr = var_importance_metric_map.find(upper_var_importance_metric_str);
66  if (itr == var_importance_metric_map.end()) {
68  }
69  return itr->second;
70 }
std::string to_upper(const std::string &str)

+ Here is the call graph for this function:

+ Here is the caller graph for this function: