Keras：从ImageDataGenerator或predict_generator获取True标签（y_test）

Question

我ImageDataGenerator().flow_from_directory(...)用来从目录中生成一批数据。

在模型成功构建之后，我想获得两列True和Predicted类标签的数组。有了model.predict_generator(validation_generator, steps=NUM_STEPS)我，我可以得到一个预测类的numpy数组。是否可以使predict_generator输出具有相应的True类标签？

要添加：validation_generator.classes确实会打印True标签，但按照从目录中检索标签的顺序，它不会考虑通过扩增进行的批处理或样品扩展。

Answer 1

您可以通过以下方式获取预测标签：

 y_pred = numpy.rint(predictions)

您可以通过以下方式获得真实的标签：

y_true = validation_generator.classes

您应该shuffle=False在此之前在验证生成器中进行设置。

最后，您可以通过以下方式打印混淆矩阵

print confusion_matrix(y_true, y_pred)