model_family#

Enum for family of machine learning models.

Module Contents#

Classes Summary#

ModelFamily

Enum for family of machine learning models.

Contents#

class evalml.model_family.model_family.ModelFamily[source]#

Enum for family of machine learning models.

Attributes

ARIMA

ARIMA model family.

BASELINE

Baseline model family.

CATBOOST

CatBoost model family.

DECISION_TREE

Decision Tree model family.

ENSEMBLE

Ensemble model family.

EXPONENTIAL_SMOOTHING

Exponential Smoothing model family.

EXTRA_TREES

Extra Trees model family.

K_NEIGHBORS

K Nearest Neighbors model family.

LIGHTGBM

LightGBM model family.

LINEAR_MODEL

Linear model family.

NONE

None

PROPHET

Prophet model family.

RANDOM_FOREST

Random Forest model family.

SVM

SVM model family.

VOWPAL_WABBIT

Vowpal Wabbit model family.

XGBOOST

XGBoost model family.

Methods

is_tree_estimator

Checks whether the estimator's model family uses trees.

name

The name of the Enum member.

value

The value of the Enum member.

is_tree_estimator(self)[source]#

Checks whether the estimator’s model family uses trees.

name(self)#

The name of the Enum member.

value(self)#

The value of the Enum member.