evalml.pipelines.components.utils.get_estimators¶
-
evalml.pipelines.components.utils.get_estimators(problem_type, model_families=None)[source]¶ Returns the estimators allowed for a particular problem type.
Can also optionally filter by a list of model types.
- Parameters
problem_type (ProblemTypes or str) – problem type to filter for
model_families (list[ModelFamily] or list[str]) – model families to filter for
- Returns
a list of estimator subclasses
- Return type
list[class]