def plot_confusion_matrix(cm, target_names, title='Confusion matrix', cmap=plt.cm.cool, normalize=True):
时间: 2024-06-02 13:08:30 浏览: 146
Confusion Matrix in Python plot a pretty confusion matrix
这是一个用于绘制混淆矩阵的函数,其中包括以下参数:
- `cm`: 混淆矩阵,通常是一个二维数组,矩阵的每个元素代表真实标签和预测标签的交叉计数。
- `target_names`: 类别名称,代表每个类别的名称。
- `title`: 图表的标题。
- `cmap`: 用于绘制矩阵的颜色图谱。
- `normalize`: 是否将混淆矩阵归一化为百分比形式。
函数会绘制出一张混淆矩阵的热力图,可以直观地显示出模型预测的准确性和错误情况。
阅读全文