utils#

EvalML 中用于数据检查的实用方法。

模块内容#

函数#

handle_data_check_action_code

通过返回 DataCheckActionCode 或从字符串转换来处理数据检查操作码。

内容#

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

通过返回 DataCheckActionCode 或从字符串转换来处理数据检查操作码。

参数

action_code (str 或 DataCheckActionCode) – 需要处理的数据检查操作码。

返回

DataCheckActionCode 枚举

引发

KeyError – 如果输入不是有效的 DataCheckActionCode 枚举值。
ValueError – 如果输入不是字符串或 DataCheckActionCode 对象。

示例

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