首页 > 解决方案 > 如何更改此混淆矩阵中的字体大小?

问题描述

我有绘制混淆矩阵的功能

import itertools
import matplotlib.pyplot as plt

def plot_confusion_matrix(cm,classes,normalize=False,title='Confusion matrix',cmap=plt.cm.Blues):

   plt.figure(figsize=(15,10))

   plt.imshow(cm,interpolation='nearest',cmap=cmap)
   plt.title(title)
   plt.colorbar()

   tick_marks=np.arange(len(classes))
   plt.xticks(tick_marks,classes,rotation=45)
   plt.yticks(tick_marks,classes)

   if normalize:
    cm=cm.astype('float')/cm.sum(axis=1)[:,np.newaxis]
    cm=np.around(cm,decimals=2)
    cm[np.isnan(cm)]=0.0
    print('Normalized confusion matrix')

   else:
    print('Confusion matrix, without normalization')


   thresh=cm.max()/2

   for i, j in itertools.product(range(cm.shape[0]), range(cm.shape[1])):

    plt.text(j, i, cm[i, j],

             horizontalalignment="center",

             color="white" if cm[i, j] > thresh else "black")

    plt.tight_layout()

    plt.ylabel('True label')

    plt.xlabel('Predicted label')

然后,实现功能

cm=confusion_matrix(validation_generator.classes,y_pred)
plot_confusion_matrix(cm,target_names,title='confusion matrix')

输出这张图片 在此处输入图片描述

如何增加图像中数字的大小?

标签: matplotlibconfusion-matrix

解决方案


要更改 matplotlib 图的字体大小,您可以尝试这种方法 -

    plt.rcParams.update({'font.size': 22})

此外,您还可以在此处探索其他格式选项


推荐阅读