Model Family

Family of machine learning models.

Submodules

• model_family
• utils

Package Contents

Classes Summary

ModelFamily

Enum for family of machine learning models.

Functions

handle_model_family

Handles model_family by either returning the ModelFamily or converting from a string.

Contents

evalml.model_family.handle_model_family(model_family)

Handles model_family by either returning the ModelFamily or converting from a string.

Parameters

model_family (str or ModelFamily) – Model type that needs to be handled.

Returns

ModelFamily

Raises

• KeyError – If input is not a valid model family.

• ValueError – If input is not a string or ModelFamily object.

class evalml.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.
VARMAX	VARMAX 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.