evalml.pipelines.components.utils.allowed_model_families¶
-
evalml.pipelines.components.utils.allowed_model_families(problem_type)[source]¶ List the model types allowed 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]