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

Question

我收到此错误：

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)))

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

Answer 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)))