模型家族

机器学习模型的家族。

子模块

• model_family
• utils

包内容

类摘要

ModelFamily

机器学习模型家族的枚举。

函数

handle_model_family

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

目录

evalml.model_family.handle_model_family(model_family)

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

参数

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

返回值

ModelFamily

抛出

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

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

class evalml.model_family.ModelFamily

机器学习模型家族的枚举。

属性

ARIMA	ARIMA 模型家族。
BASELINE	基线模型家族。
CATBOOST	CatBoost 模型家族。
DECISION_TREE	决策树模型家族。
ENSEMBLE	集成模型家族。
EXPONENTIAL_SMOOTHING	指数平滑模型家族。
EXTRA_TREES	极端随机树模型家族。
K_NEIGHBORS	K 近邻模型家族。
LIGHTGBM	LightGBM 模型家族。
LINEAR_MODEL	线性模型家族。
NONE	无
PROPHET	Prophet 模型家族。
RANDOM_FOREST	随机森林模型家族。
SVM	SVM 模型家族。
VARMAX	VARMAX 模型家族。
VOWPAL_WABBIT	Vowpal Wabbit 模型家族。
XGBOOST	XGBoost 模型家族。

方法

is_tree_estimator	检查估计器的模型家族是否使用树模型。
name	枚举成员的名称。
value	枚举成员的值。

is_tree_estimator(self)

检查估计器的模型家族是否使用树模型。

name(self)

枚举成员的名称。

value(self)

枚举成员的值。