AttributeError：“NumpyArrayIterator”对象没有属性“类”

Posted 2023-03-12

【中文标题】AttributeError：“NumpyArrayIterator”对象没有属性“类”

【英文标题】：AttributeError: 'NumpyArrayIterator' object has no attribute 'classes'

【发布时间】：2020-02-25 16:26:51

【问题描述】：

我收到此错误：

AttributeError: 'NumpyArrayIterator' 对象没有属性 'classes'

我正在尝试制作一个混淆矩阵来评估我训练过的神经网络。我在 fit_generator 函数进行训练之前使用 ImageDatagenerator 和 datagen.flow 函数。

对于预测，我在测试集上使用 predict_generator 函数。到目前为止一切正常。问题出现如下：

test_generator.reset()
pred = model.predict_generator(test_generator, steps=len(test_generator), verbose=2)

from sklearn.metrics import classification_report, confusion_matrix, cohen_kappa_score

y_pred = np.argmax(pred, axis=1)

print('Confusion Matrix')
print(pd.DataFrame(confusion_matrix(test_generator.classes, y_pred)))

我应该看到一个混淆矩阵，但我看到了一个错误。在运行实际数据集之前，我使用示例数据运行了相同的代码，并且确实显示了结果。

【问题讨论】：

如果我理解得很好，你想制作test_generator类和预测（y_pred）类的confusion_matrix？

请。真值和预测值的混淆矩阵。

【参考方案1】：

首先您需要从生成器中提取标签，然后将它们放入confusion_matrix 函数中。 要提取标签，请使用x_gen,y_gen = test_generator.next()，只需注意标签是一种热编码。 示例：

test_generator.reset()
pred = model.predict_generator(test_generator, steps=len(test_generator), verbose=2)

from sklearn.metrics import classification_report, confusion_matrix, cohen_kappa_score

y_pred = np.argmax(pred, axis=1)

x_gen,y_gen = test_generator.next()
y_gen = np.argmax(y_gen, axis=1)

print('Confusion Matrix')
print(pd.DataFrame(confusion_matrix(y_gen, y_pred)))

【讨论】：

这给了我以下错误：ValueError: Found input variables with contrast numbers of samples: [32, 8871] y_gen 也应该是 8871。