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