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::MLPACK, MLFramework::INVALID }
enum	KMeansInitStrategy {   KMeansInitStrategy::DEFAULT, KMeansInitStrategy::DETERMINISTIC, KMeansInitStrategy::RANDOM, KMeansInitStrategy::PLUS_PLUS,   KMeansInitStrategy::INVALID }

Functions
MLFramework	get_ml_framework (const std::string &ml_framework_str)
KMeansInitStrategy	get_kmeans_init_type (const std::string &init_type_str)

Enumeration Type Documentation

enum KMeansInitStrategy

strong

Enumerator
DEFAULT
DETERMINISTIC
RANDOM
PLUS_PLUS
INVALID

Definition at line 38 of file MLTableFunctionsCommon.h.

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

enum MLFramework

strong

Enumerator
DEFAULT
ONEDAL
MLPACK
INVALID

Definition at line 23 of file MLTableFunctionsCommon.h.

{ DEFAULT, ONEDAL, MLPACK, INVALID };

Function Documentation

KMeansInitStrategy get_kmeans_init_type

(

const std::string &

init_type_str

)

Definition at line 40 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

)

Definition at line 25 of file MLTableFunctionsCommon.h.

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

Referenced by dbscan__cpu_template(), kmeans__cpu_template(), linear_reg_fit__cpu_template(), and linear_reg_predict__cpu_template().

                                                                {
  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},
      {"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: