evalml.pipelines.get_estimators

evalml.pipelines.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]