EA ModelTask Fix (#2316)

erroranalysis/erroranalysis/_internal/constants.py

@@ -63,6 +63,8 @@ class ModelTask(str, Enum):
 
     CLASSIFICATION = 'classification'
     REGRESSION = 'regression'
+    IMAGE_CLASSIFICATION = 'image_classification'
+    MULTILABEL_IMAGE_CLASSIFICATION = 'multilabel_image_classification'
     OBJECT_DETECTION = 'object_detection'
     UNKNOWN = 'unknown'
     Classification = 'classification'

erroranalysis/erroranalysis/analyzer/error_analyzer.py

@@ -105,8 +105,11 @@ class BaseAnalyzer(ABC):
         self._categories = []
         self._categorical_indexes = []
         self._category_dictionary = {}
-        self._model_task = model_task
         self._classes = classes
+        if model_task == ModelTask.IMAGE_CLASSIFICATION or \
+           model_task == ModelTask.MULTILABEL_IMAGE_CLASSIFICATION:
+            model_task = ModelTask.CLASSIFICATION
+        self._model_task = model_task
         if model_task == ModelTask.CLASSIFICATION:
             if metric is None:
                 metric = Metrics.ERROR_RATE