Keras:从flow_from_directory获取图像和标签数组

时间:2017-10-05 12:12:20

标签: python tensorflow keras

我在训练模型后尝试使用scikit获取混淆矩阵,但由于我使用了flow_from_directory,我无法访问数据和标签,或者我不知道知道这样做的方法。由于使用了scikit混淆矩阵方法,如:

confusion_matrix(y_true, y_pred)

并且flow_from_directory没有返回真正的标签。有没有办法直接从flow_from_directory或任何其他方法获取它们?

1 个答案:

答案 0 :(得分:5)

对于任何generator,您可以像任何可迭代的那样生成数据:

for x, y in something.flow_from_directory(...):
    #do stuff

小心决定何时停止。用于训练keras的发生器是无限的。

相关问题