我在用ImageDataGenerator().flow_from_directory(...)从目录生成批量数据。
模型成功构建后，我想获得真实和预测类标签的两列数组。和model.predict_generator(validation_generator, steps=NUM_STEPS)我可以获得预测类的 numpy 数组。是否有可能拥有predict_generator输出对应的True类标签？
要添加：validation_generator.classes 确实打印 True 标签，但按照从目录中检索它们的顺序，它不考虑批处理或通过增强进行的样本扩展。
您可以通过以下方式获取预测标签：
y_pred = numpy.rint(predictions)
您可以通过以下方式获得真正的标签：
y_true = validation_generator.classes
你应该设置shuffle=False在此之前的验证生成器中。
最后，您可以通过打印混淆矩阵
print confusion_matrix(y_true, y_pred)
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
