MLTableFunctionsCommon.h File Reference

#include "QueryEngine/TableFunctions/SystemFunctions/os/Shared/TableFunctionsCommon.hpp"
#include <map>

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::ONEAPI, 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 39 of file MLTableFunctionsCommon.h.

{ DEFAULT, DETERMINISTIC, RANDOM, PLUS_PLUS, INVALID };

enum MLFramework

strong

Enumerator
DEFAULT
ONEDAL
ONEAPI
MLPACK
INVALID

Definition at line 23 of file MLTableFunctionsCommon.h.

{ DEFAULT, ONEDAL, ONEAPI, MLPACK, INVALID };

enum VarImportanceMetric

strong

Enumerator
DEFAULT
NONE
MDI
MDA
MDA_SCALED
INVALID

Definition at line 55 of file MLTableFunctionsCommon.h.

{ DEFAULT, NONE, MDI, MDA, MDA_SCALED, INVALID };

Function Documentation

KMeansInitStrategy get_kmeans_init_type ( const std::string &  init_type_str )

inline

Definition at line 41 of file MLTableFunctionsCommon.h.

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

Referenced by kmeans__cpu_template().

                                                                               {
  const auto upper_init_type_str = to_upper(init_type_str);
  const static std::map<std::string, KMeansInitStrategy> kmeans_init_type_map = {
      {"DEFAULT", KMeansInitStrategy::DEFAULT},
      {"DETERMINISTIC", KMeansInitStrategy::DETERMINISTIC},
      {"RANDOM", KMeansInitStrategy::RANDOM},
      {"PLUS_PLUS", KMeansInitStrategy::PLUS_PLUS}};
  const auto itr = kmeans_init_type_map.find(upper_init_type_str);
  if (itr == kmeans_init_type_map.end()) {
    return KMeansInitStrategy::INVALID;
  }
  return itr->second;
}

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, ONEAPI, 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().

                                                                       {
  const auto upper_ml_framework_str = to_upper(ml_framework_str);
  const static std::map<std::string, MLFramework> ml_framework_map = {
      {"DEFAULT", MLFramework::DEFAULT},
      {"ONEDAL", MLFramework::ONEDAL},
      {"ONEAPI", MLFramework::ONEAPI},
      {"MLPACK", MLFramework::MLPACK}};
  const auto itr = ml_framework_map.find(upper_ml_framework_str);
  if (itr == ml_framework_map.end()) {
    return MLFramework::INVALID;
  }
  return itr->second;
}

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 57 of file MLTableFunctionsCommon.h.

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

Referenced by random_forest_reg_fit_impl().

                                                {
  const auto upper_var_importance_metric_str = to_upper(var_importance_metric_str);
  const static std::map<std::string, VarImportanceMetric> var_importance_metric_map = {
      {"DEFAULT", VarImportanceMetric::DEFAULT},
      {"NONE", VarImportanceMetric::NONE},
      {"MDI", VarImportanceMetric::MDI},
      {"MDA", VarImportanceMetric::MDA},
      {"MDA_SCALED", VarImportanceMetric::MDA_SCALED}};
  const auto itr = var_importance_metric_map.find(upper_var_importance_metric_str);
  if (itr == var_importance_metric_map.end()) {
    return VarImportanceMetric::INVALID;
  }
  return itr->second;
}

Here is the call graph for this function:

Here is the caller graph for this function: