evalml.model_family.ModelFamily¶

class evalml.model_family.ModelFamily(value)[source]¶

Enum for family of machine learning models.

Attributes

`BASELINE`	Baseline model family.
`CATBOOST`	CatBoost model family.
`DECISION_TREE`	Decision Tree model family.
`ENSEMBLE`	Ensemble model family.
`EXTRA_TREES`	Extra Trees model family.
`K_NEIGHBORS`	K Nearest Neighbors model family.
`LIGHTGBM`	LightGBM model family.
`LINEAR_MODEL`	Linear model family.
`NONE`	None
`RANDOM_FOREST`	Random Forest model family.
`SVM`	SVM model family.
`XGBOOST`	XGBoost model family.