#320 related. #363 temp added, todo. #355 covered for 3 layers (temp)

nicolay-r · Jul 22, 2022 · 976dee5 · 976dee5
1 parent 231baa5
commit 976dee5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arekit/contrib/utils/evaluation/evaluators/three_class.py b/arekit/contrib/utils/evaluation/evaluators/three_class.py
@@ -3,14 +3,14 @@
 from arekit.contrib.utils.evaluation.results.three_class_prf import ThreeClassPrecRecallF1EvalResult
 
 
-class TwoClassEvaluator(BaseEvaluator):
+class ThreeClassEvaluator(BaseEvaluator):
 
     def __init__(self, comparator, label1, label2, label3, get_item_label_func):
         assert(isinstance(label1, Label))
         assert(isinstance(label2, Label))
         assert(isinstance(label3, Label))
         assert(callable(get_item_label_func))
-        super(TwoClassEvaluator, self).__init__(comparator)
+        super(ThreeClassEvaluator, self).__init__(comparator)
 
         self.__label1 = label1
         self.__label2 = label2