pytorch 绘制混淆矩阵

#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