fix : minor eidt in weighted_kappa method #243

sepandhaghighi · Apr 22, 2020 · 8a6c073 · 8a6c073
1 parent e4c58d9
commit 8a6c073
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pycm/pycm_obj.py b/pycm/pycm_obj.py
@@ -695,7 +695,7 @@ def weighted_average(self, param, weight=None, none_omit=False):
         except Exception:
             return "None"
 
-    def weighted_kappa(self,weight):
+    def weighted_kappa(self,weight=None):
         """
         Calculate weighted kappa.
 
@@ -705,8 +705,8 @@ def weighted_kappa(self,weight):
         """
         if matrix_check(weight) is False:
             warn(WEIGHTED_KAPPA_WARNING, RuntimeWarning)
-            return cm.Kappa
+            return self.Kappa
         if set(weight.keys()) != set(self.classes):
             warn(WEIGHTED_KAPPA_WARNING, RuntimeWarning)
-            return cm.Kappa
+            return self.Kappa
         return weighted_kappa_calc(self.classes,self.table,self.P,self.TOP,self.POP,weight)