evalml.pipelines.get_pipelines

evalml.pipelines.get_pipelines(problem_type, model_families=None)

Returns the pipelines 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 pipeline classes

Return type

list[PipelineBase]