Model Family¶
Family of machine learning models.
Submodules¶
Package Contents¶
Classes Summary¶
Enum for family of machine learning models. |
Functions¶
Handles model_family by either returning the ModelFamily or converting from a string. |
Contents¶
-
evalml.model_family.
handle_model_family
(model_family)[source]¶ Handles model_family by either returning the ModelFamily or converting from a string.
- Parameters
model_family (str or ModelFamily) – Model type that needs to be handled.
- Returns
ModelFamily
- Raises
KeyError – If input is not a valid model family.
ValueError – If input is not a string or ModelFamily object.
-
class
evalml.model_family.
ModelFamily
[source]¶ Enum for family of machine learning models.
Attributes
ARIMA
ARIMA model family.
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
PROPHET
Prophet model family.
RANDOM_FOREST
Random Forest model family.
SVM
SVM model family.
VOWPAL_WABBIT
Vowpal Wabbit model family.
XGBOOST
XGBoost model family.
Methods
Checks whether the estimator’s model family uses trees.
The name of the Enum member.
The value of the Enum member.
-
name
(self)¶ The name of the Enum member.
-
value
(self)¶ The value of the Enum member.
-