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，你可以得到的數據與任何可迭代：

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

要小心，以決定什麼時候停止。 keras培訓的發電機是無限的。