"cnn: [7, 1, 2, 0, 0, 0, 1, 0, 7, 0, 0, 1, 0, 0, 0, 7, 0, 0, 7, 0, 2, 0, 0, 2, 0, 0, 7, 2, 2, 0, 7, 7] gru: [7, 1, 0, 0, 0, 0, 0, 0, 7, 0, 0, 1, 0, 0, 0, 7, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 7, 2, 2, 0, 7, 7] labels :[7, 1, 0, 0, 0, 0, 0, 0, 7, 0, 0, 1, 0, 0, 0, 7, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 7, 7] 有很多条cnn和gru模型的八分类器的分类结果,格式如上,labels是原始标签,根据labels求出混淆矩阵内容:混淆矩阵是一种用于可视化分类模型性能的工具,它可以展示模型在每个类别上的预测结果与实际标签之间的关系。混淆矩阵的行表示实际标签,列表示预测结果。

首先,我们需要将cnn和gru的分类结果与标签进行比较,计算混淆矩阵。以下是一个示例代码,用于计算混淆矩阵:

import numpy as np
from sklearn.metrics import confusion_matrix

cnn_result = [7, 1, 2, 0, 0, 0, 1, 0, 7, 0, 0, 1, 0, 0, 0, 7, 0, 0, 7, 0, 2, 0, 0, 2, 0, 0, 7, 2, 2, 0, 7, 7]
gru_result = [7, 1, 0, 0, 0, 0, 0, 0, 7, 0, 0, 1, 0, 0, 0, 7, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 7, 2, 2, 0, 7, 7]
labels = [7, 1, 0, 0, 0, 0, 0, 0, 7, 0, 0, 1, 0, 0, 0, 7, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 7, 7]

# 将结果转为numpy数组
cnn_result = np.array(cnn_result)
gru_result = np.array(gru_result)
labels = np.array(labels)

# 计算混淆矩阵
cnn_confusion_matrix = confusion_matrix(labels, cnn_result)
gru_confusion_matrix = confusion_matrix(labels, gru_result)

print("CNN混淆矩阵:")
print(cnn_confusion_matrix)

print("GRU混淆矩阵:")
print(gru_confusion_matrix)

运行以上代码,可以得到CNN和GRU模型的混淆矩阵。混淆矩阵将会显示每个类别的预测结果和实际标签之间的对应关系。


原文地址: https://www.cveoy.top/t/topic/qC2z 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录