evalml.model_family.ModelFamily

class evalml.model_family.ModelFamily(value)[source]

Enum for family of machine learning models.

Attributes

BASELINE

Baseline model family.

CATBOOST

CatBoost model family.

DECISION_TREE

Decision Tree model family.

ENSEMBLE

Ensemble 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

RANDOM_FOREST

Random Forest model family.

SVM

SVM model family.

XGBOOST

XGBoost model family.