utils

Utility methods for EvalML’s model families.

Module Contents

Functions

handle_model_family

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

Contents

evalml.model_family.utils.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.

model_family Model Understanding