evalml.pipelines.utils.get_estimators

evalml.pipelines.utils.get_estimators(problem_type, model_families=None)

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]

evalml.pipelines.graph_permutation_importance evalml.pipelines.utils.make_pipeline