fix : minor bug in Compare fixed #347

sepandhaghighi · Oct 10, 2021 · 63f9402 · 63f9402
1 parent 646857a
commit 63f9402
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycm/pycm_compare.py b/pycm/pycm_compare.py
@@ -53,7 +53,7 @@ def __init__(self, cm_dict, by_class=False, class_weight=None, class_benchmark_w
         self.scores = None
         self.sorted = None
         self.classes = None
-        __compare_assign_handler__(self, cm_dict, class_weight, digit)
+        __compare_assign_handler__(self, cm_dict, class_weight, class_benchmark_weight, overall_benchmark_weight, digit)
         __compare_class_handler__(self, cm_dict)
         __compare_overall_handler__(self, cm_dict)
         __compare_rounder__(self, cm_dict)