gpt4 book ai didi

Keras:从 ImageDataGenerator 或 Predict_generator 获取真实标签 (y_test)

转载 作者:行者123 更新时间:2023-12-02 10:17:57 24 4
gpt4 key购买 nike

我正在使用 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

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

打印confusion_matrix(y_true, y_pred)

关于Keras:从 ImageDataGenerator 或 Predict_generator 获取真实标签 (y_test),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45413712/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com