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

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)

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.