ClassificationModels Class

public final class ClassificationModels
extends ExpandableStringEnum<ClassificationModels>

Enum for all classification models supported by AutoML.

Field Summary

Modifier and Type Field and Description
static final ClassificationModels BERNOULLI_NAIVE_BAYES

Static value BernoulliNaiveBayes for ClassificationModels.

static final ClassificationModels DECISION_TREE

Static value DecisionTree for ClassificationModels.

static final ClassificationModels EXTREME_RANDOM_TREES

Static value ExtremeRandomTrees for ClassificationModels.

static final ClassificationModels GRADIENT_BOOSTING

Static value GradientBoosting for ClassificationModels.

static final ClassificationModels KNN

Static value KNN for ClassificationModels.

static final ClassificationModels LIGHT_GBM

Static value LightGBM for ClassificationModels.

static final ClassificationModels LINEAR_SVM

Static value LinearSVM for ClassificationModels.

static final ClassificationModels LOGISTIC_REGRESSION

Static value LogisticRegression for ClassificationModels.

static final ClassificationModels MULTINOMIAL_NAIVE_BAYES

Static value MultinomialNaiveBayes for ClassificationModels.

static final ClassificationModels RANDOM_FOREST

Static value RandomForest for ClassificationModels.

static final ClassificationModels SGD

Static value SGD for ClassificationModels.

static final ClassificationModels SVM

Static value SVM for ClassificationModels.

static final ClassificationModels XGBOOST_CLASSIFIER

Static value XGBoostClassifier for ClassificationModels.

Constructor Summary

Constructor Description
ClassificationModels()

Deprecated

Use the fromString(String name) factory method.

Creates a new instance of ClassificationModels value.

Method Summary

Modifier and Type Method and Description
static ClassificationModels fromString(String name)

Creates or finds a ClassificationModels from its string representation.

static Collection<ClassificationModels> values()

Gets known ClassificationModels values.

Methods inherited from ExpandableStringEnum

Methods inherited from java.lang.Object

Field Details

BERNOULLI_NAIVE_BAYES

public static final ClassificationModels BERNOULLI_NAIVE_BAYES

Static value BernoulliNaiveBayes for ClassificationModels.

DECISION_TREE

public static final ClassificationModels DECISION_TREE

Static value DecisionTree for ClassificationModels.

EXTREME_RANDOM_TREES

public static final ClassificationModels EXTREME_RANDOM_TREES

Static value ExtremeRandomTrees for ClassificationModels.

GRADIENT_BOOSTING

public static final ClassificationModels GRADIENT_BOOSTING

Static value GradientBoosting for ClassificationModels.

KNN

public static final ClassificationModels KNN

Static value KNN for ClassificationModels.

LIGHT_GBM

public static final ClassificationModels LIGHT_GBM

Static value LightGBM for ClassificationModels.

LINEAR_SVM

public static final ClassificationModels LINEAR_SVM

Static value LinearSVM for ClassificationModels.

LOGISTIC_REGRESSION

public static final ClassificationModels LOGISTIC_REGRESSION

Static value LogisticRegression for ClassificationModels.

MULTINOMIAL_NAIVE_BAYES

public static final ClassificationModels MULTINOMIAL_NAIVE_BAYES

Static value MultinomialNaiveBayes for ClassificationModels.

RANDOM_FOREST

public static final ClassificationModels RANDOM_FOREST

Static value RandomForest for ClassificationModels.

SGD

public static final ClassificationModels SGD

Static value SGD for ClassificationModels.

SVM

public static final ClassificationModels SVM

Static value SVM for ClassificationModels.

XGBOOST_CLASSIFIER

public static final ClassificationModels XGBOOST_CLASSIFIER

Static value XGBoostClassifier for ClassificationModels.

Constructor Details

ClassificationModels

@Deprecated
public ClassificationModels()

Deprecated

Use the fromString(String name) factory method.

Creates a new instance of ClassificationModels value.

Method Details

fromString

public static ClassificationModels fromString(String name)

Creates or finds a ClassificationModels from its string representation.

Parameters:

name - a name to look for.

Returns:

the corresponding ClassificationModels.

values

public static Collection<ClassificationModels> values()

Gets known ClassificationModels values.

Returns:

known ClassificationModels values.

Applies to