evalml.pipelines.components.utils.allowed_model_families

evalml.pipelines.components.utils.allowed_model_families(problem_type)

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]

evalml.pipelines.components.Estimator.save evalml.pipelines.components.utils.get_estimators