绘制混淆矩阵，计算kappa系数

import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
#绘制混淆矩阵
conmatrix=np.array([[152,1,2,0,0,2,0,0,0],
 [1,28,0,0,0,0,0,0,0],
 [5,0,56,0,0,3,0,0,0],
 [2,0,0,14,0,1,0,0,0],
 [2,0,0,0,24,0,0,0,0],
 [9,1,3,0,0,85,1,0,0],
 [2,0,0,0,0,1,11,0,0],
 [0,0,0,0,0,0,0,6,0],
 [4,0,0,0,0,1,0,0,15]])
df_cm=pd.DataFrame(conmatrix,
                  columns=['Estate','Tourist','Business','Medical','Railway','Road','Education','Catering','Shopping'])#&Transport

plt.figure(figsize=(10,8))
f, ax= plt.subplots(figsize = (10, 8))
 
h=sns.heatmap(df_cm,annot=True,cmap="binary",fmt='.20g',ax=ax,linewidths=2,
            annot_kws={'size':30},cbar=False)
 
#h=sns.heatmap(data, annot=True,fmt='d',linewidths=0.5,cbar=False) #设置不使用其默认自带的colorbar
cb=h.figure.colorbar(h.collections[0]) #显示colorbar
cb.ax.tick_params(labelsize=24) #设置colorbar刻度字体大小。

ax.set_xticklabels(ax.get_xticklabels(), rotation=90)

ax.tick_params(labelsize=24)
ax.set_yticklabels(ax.get_yticklabels(), rotation=0)
 
plt.ylabel('Actual Result',fontsize=24)
plt.xlabel('Forecast Result',fontsize=24)
plt.savefig('cnn.svg',format='svg')
plt.show()

def kappa(matrix):
    n = np.sum(matrix)
    sum_po = 0
    sum_pe = 0
    for i in range(len(matrix[0])):
        sum_po += matrix[i][i]
        row = np.sum(matrix[i, :])
        col = np.sum(matrix[:, i])
        sum_pe += row * col
    po = sum_po / n
    pe = sum_pe / (n * n)
    # print(po, pe)
    return (po - pe) / (1 - pe)