#355. fixed input params.

nicolay-r · Jul 22, 2022 · 1991c33 · 1991c33
1 parent 976dee5
commit 1991c33
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/arekit/contrib/utils/evaluation/evaluators/three_class.py b/arekit/contrib/utils/evaluation/evaluators/three_class.py
@@ -14,10 +14,11 @@ def __init__(self, comparator, label1, label2, label3, get_item_label_func):
 
         self.__label1 = label1
         self.__label2 = label2
+        self.__label3 = label3
         self.__get_item_label_func = get_item_label_func
 
     def _create_eval_result(self):
         return ThreeClassPrecRecallF1EvalResult(label1=self.__label1,
                                                 label2=self.__label2,
-                                                label3=self.__label2,
+                                                label3=self.__label3,
                                                 get_item_label_func=self.__get_item_label_func)