evalml.pipelines.list_model_families

evalml.pipelines.list_model_families(problem_type)

List model type for a particular problem type

Parameters

problem_types (ProblemTypes or str) – binary, multiclass, or regression

Returns

a list of model families

Return type

list[ModelFamily]