evalml.pipelines.get_estimators

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