utils

EvalML 模型族的实用方法。

模块内容

函数

handle_model_family

通过返回 ModelFamily 或从字符串转换来处理 model_family。

内容

evalml.model_family.utils.handle_model_family(model_family)

通过返回 ModelFamily 或从字符串转换来处理 model_family。

参数

model_family (str 或 ModelFamily) – 需要处理的模型类型。

返回

ModelFamily

抛出

• KeyError – 如果输入不是有效的模型族。

• ValueError – 如果输入不是字符串或 ModelFamily 对象。