小编col*_*tor的帖子

Keras:从ImageDataGenerator或predict_generator获取True标签(y_test)

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

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

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

keras

7
推荐指数
1
解决办法
4999
查看次数

标签 统计

keras ×1