ClassificationModels Enum

Enum for all classification models supported by AutoML.

Inheritance
builtins.str
ClassificationModels
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