evalml.model_family.list_model_families

evalml.model_family.list_model_families(problem_type)[source]

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]