ClassificationModels Enum

Enum for all classification models supported by AutoML.

Inheritance

builtins.str

ClassificationModels

Enum

ClassificationModels

Constructor

ClassificationModels(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

Fields

BERNOULLI_NAIVE_BAYES

DECISION_TREE

EXTREME_RANDOM_TREES

GRADIENT_BOOSTING

KNN

LIGHT_GBM

LINEAR_SVM

LOGISTIC_REGRESSION

MULTINOMIAL_NAIVE_BAYES

RANDOM_FOREST

SGD

SVM

XG_BOOST_CLASSIFIER