fix : error tests updated #347

sepandhaghighi · Nov 28, 2021 · 84c54cd · 84c54cd
1 parent 0d4340f
commit 84c54cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Test/error_test.py b/Test/error_test.py
@@ -83,11 +83,11 @@
 pycm.pycm_obj.pycmCompareError: The domain of all ConfusionMatrix objects must be same! The sample size or the number of classes are different.
 >>> y_pred = [0, 0, 2, 1, 0, 2, 1, 0, 2, 0, 2, 1]
 >>> cm3 = ConfusionMatrix(y_actu,y_pred)
->>> cp = Compare({"cm1":cm,"cm2":cm3},by_class=True,weight={1:1,2:1})
+>>> cp = Compare({"cm1":cm,"cm2":cm3},by_class=True,class_weight={1:1,2:1})
 Traceback (most recent call last):
         ...
 pycm.pycm_obj.pycmCompareError: The class_weight type must be dictionary and also must be specified for all of the classes.
->>> cp = Compare({"cm1":cm,"cm2":cm3},by_class=True,weight=[])
+>>> cp = Compare({"cm1":cm,"cm2":cm3},by_class=True,class_weight=[])
 Traceback (most recent call last):
         ...
 pycm.pycm_obj.pycmCompareError: The class_weight type must be dictionary and also must be specified for all of the classes.