Model Family

Family of machine learning models.

Submodules

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)[source]

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[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.

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.