evalml.model_family.handle_model_family

evalml.model_family.handle_model_family(model_family)

Handles model_family by either returning the ModelFamily or converting from a string

Parameters

model_family (str or ModelFamily) – Model type that needs to be handled

Returns

ModelFamily

evalml.problem_types.ProblemTypes evalml.model_family.ModelFamily