我在调用matplotlib包画交叉矩阵时,出现如下错误。
相关代码如下:
# 定义混淆矩阵画图函数
def plot_confusion_matrix(cm, # 混淆矩阵
target_names, # 标签名
title='Confusion matrix', # 设置矩阵名称
cmap='Blues', # 这个地方设置混淆矩阵的颜色主题,这个主题看着就干净
normalize=True):
accuracy = np.trace(cm) / float(np.sum(cm)) # 计算准确率
misclass = 1 - accuracy # 计算错误率
if cmap is None:
cmap = plt.get_cmap('Blues') # 如果没有设置主题颜色,就设置主题颜色为'Blues'
plt.figure(figsize=(9, 7)) # 设置图的尺寸大小
plt.imshow(cm