Keras：從flow_from_directory獲取圖像和標簽數組

Question

我在訓練模型后試圖使用scikit獲取混淆矩陣，但由於我使用了flow_from_directory ，我無法訪問數據和標簽，或者我不知道這樣做的方法。 由於scikit混淆矩陣方法使用如下：

confusion_matrix(y_true, y_pred)

和flow_from_directory不返回真實標簽。 有沒有辦法直接從flow_from_directory或任何其他方法獲取它們？

Answer 1

對於任何generator ，您可以像任何可迭代的那樣generator數據：

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

小心決定何時停止。 用於訓練keras的發生器是無限的。