utils¶

Utility methods for the data checks in EvalML.

Module Contents¶

Functions¶

handle_data_check_action_code

Handles data check action codes by either returning the DataCheckActionCode or converting from a str.

Contents¶

evalml.data_checks.utils.handle_data_check_action_code(action_code)[source]¶

Handles data check action codes by either returning the DataCheckActionCode or converting from a str.

Parameters

action_code (str or DataCheckActionCode) – Data check action code that needs to be handled.

Returns

DataCheckActionCode enum

Raises

KeyError – If input is not a valid DataCheckActionCode enum value.
ValueError – If input is not a string or DatCheckActionCode object.

Examples

>>> assert handle_data_check_action_code("drop_col") == DataCheckActionCode.DROP_COL
>>> assert handle_data_check_action_code("DROP_ROWS") == DataCheckActionCode.DROP_ROWS
>>> assert handle_data_check_action_code("Impute_col") == DataCheckActionCode.IMPUTE_COL

uniqueness_data_check

Demos