W3School TIY Editor

# 这三行代码使编译器支持绘图功能：

import sys

import matplotlib

matplotlib.use('Agg')

​

import matplotlib.pyplot as plt

import numpy

from sklearn import metrics

​

# 生成模拟二分类数据（正类概率 90%）：

actual = numpy.random.binomial(1, 0.9, size=1000)

predicted = numpy.random.binomial(1, 0.9, size=1000)

​

# 计算混淆矩阵：

confusion_matrix = metrics.confusion_matrix(actual, predicted)

​

# 创建混淆矩阵可视化对象：

cm_display = metrics.ConfusionMatrixDisplay(

    confusion_matrix=confusion_matrix, 

    display_labels=[0, 1]

)

​

# 绘制并显示混淆矩阵：

cm_display.plot()

plt.show()

​

# 这两行代码使编译器能够输出图形：

plt.savefig(sys.stdout.buffer)

sys.stdout.flush()