model_family

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.

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.

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.