#cm是统计好的混淆矩阵,file_path是存储混淆矩阵图像的路径,epoch为我自己传参过来的测试的次数,可以去掉epoch,看大家的要求。
import matplotlib.pyplot as plt
import numpy as np
def plot_Matrix(cm, file_path, epoch):
cmap=plt.cm.Blues
classes = ['Social', 'Using computer', 'Reading']#规定出来的x,y轴的值,
plt.rc('font',size='6') # 设置字体大小
cms = [0]*13
for j in range(13):
for i in range(13):
cms[j