modify

jamesbing · Jun 29, 2017 · 0bfb05a · 0bfb05a
1 parent 07ba227
commit 0bfb05a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hypertector/code/hypertector.py b/hypertector/code/hypertector.py
@@ -647,12 +647,12 @@ def cal_kappa(result_mat):
         pe_zong = result_matrix[0:classes,j].sum()
         pe_heng = result_matrix[j,0:classes].sum()
         pe = pe + pe_zong * pe_heng
-        sum = sum + pe_zong + pe_heng
+    sum = result_matrix.sum()
 
     for k in range(classes):
         pa = pa + result_matrix[k][k]
 
-    pe = (pe / sum) / sum
+    pe = pe / (sum * sum)
     pa = pa / sum
 
     kappa = (pa - pe)/(1-pe)