Source code for evalml.problem_types.problem_types
from enum import Enum
[docs]class ProblemTypes(Enum):
"""Enum for type of machine learning problem: BINARY, MULTICLASS, or REGRESSION"""
BINARY = 'binary'
MULTICLASS = 'multiclass'
REGRESSION = 'regression'
def __str__(self):
problem_type_dict = {ProblemTypes.BINARY.name: "Binary Classification",
ProblemTypes.MULTICLASS.name: "Multiclass Classification",
ProblemTypes.REGRESSION.name: "Regression"}
return problem_type_dict[self.name]