时间戳

首页 > 解决方案 > 测试准确度为 98% 的模型的混淆矩阵不准确

python - 测试准确度为 98% 的模型的混淆矩阵不准确

问题描述

我训练了一个二元分类模型，得到了 98% 的测试准确率和 99% 的训练准确率。

今天我想计算混淆矩阵并使用下面的代码来计算它们。

model = load_model("model.h5")

testGenerator = ImageDataGenerator(rotation_range=5,
                                width_shift_range=0.2,
                                height_shift_range=0.2,
                                horizontal_flip=False,
                                fill_mode='nearest'
                                )   

testData = testGenerator.flow_from_directory(
                                'Location', 
                                target_size=(74,448),                                                 
                                batch_size=15,
                                class_mode='binary',
                                shuffle=False
                                )

proba = model.predict_generator(testData,steps=3000//15)
y_true = np.array([0] * 1482 + [1] * 1482 )
y_pred = proba > 0.5
print(confusion_matrix(y_true, y_pred))

我收到了这个混淆矩阵：

混淆矩阵

正如 sklearn 所说：

在此处输入图像描述

这在这里说假阴性和假阳性是如此之高。既然我有 98% 的测试准确率，这怎么可能呢？此外，我多次使用该模型生成预测（使用 model.predict() 函数）并手动检查它们。但每次它都给了我正确的分类。

任何想法如何获得准确的结果？

标签： pythontensorflowscikit-learnkerasconfusion-matrix

解决方案

让我们从头开始。消息“TypeError: unhashable type: 'numpy.ndarray'”表示您不能numpy.ndarray用作字典键，因为它不是不可变对象。首先将其转换为tuple或其他内容以使其不可变。

关于您的混淆矩阵，我敢打赌生成器会以不可预知的顺序从文件夹中加载文件，但您已y_true设置为 1482zeros和 1482 ones- 这可能与生成器生成的文件的顺序匹配，也可能不匹配。因此，您会得到有趣的结果。

推荐阅读