python 绘制混淆矩阵

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python 绘制混淆矩阵相关的知识,希望对你有一定的参考价值。

%matplotlib inline
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix


def print_confusion_matrix():
    # Get the true classifications for the test-set.
    # Get the confusion matrix using sklearn.
    cls_true = []
    cls_pred = []
    cm = confusion_matrix(y_true=cls_true,
                          y_pred=cls_pred)

    # Print the confusion matrix as text.
    print(cm)

    # Plot the confusion matrix as an image.
    plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)

    # Make various adjustments to the plot.
    plt.tight_layout()
    plt.colorbar()
    tick_marks = np.arange(num_classes)
    plt.xticks(tick_marks, range(num_classes))
    plt.yticks(tick_marks, range(num_classes))
    plt.xlabel('Predicted')
    plt.ylabel('True')
    
    # Ensure the plot is shown correctly with multiple plots
    # in a single Notebook cell.
    plt.show()

以上是关于python 绘制混淆矩阵的主要内容,如果未能解决你的问题,请参考以下文章

python是不是有绘制混淆矩阵的函数,怎么来实现

如何在python中使用修改后的输出大小绘制混淆矩阵并输出为.svg图像?

如何用最少的代码在 Python 中绘制不同类型的混淆矩阵?

python绘制混淆矩阵(2s-AGCN结果分析)

Python遥感图像处理应用篇(二十八):Python绘制遥感图像分类结果混淆矩阵和计算分类精度

Python遥感图像处理应用篇(二十八):Python绘制遥感图像分类结果混淆矩阵和计算分类精度