0

我知道predict_generator輸出概率。爲了得到這個類,我只是找到最大可能性的索引,並且這將是最可能的類。不過,我發現在完成此操作之後,我得到的結果與調用predict_classes的結果不同。我不懂爲什麼。有人可以解釋這個嗎?model.predict_classes vs model.predict_generator in keras

回答

1

發電機Keras使用水珠要列出按字母順序排序,就可以得到所使用的類文件夾使用

# save classes to JSON 
class_json = json.dumps(train_generator.class_indices) 
with open("class.json", "w") as class_file: 
    class_file.write(class_json) 

樣品在批處理髮電機(在這裏)改組與這樣,當一批在訓練中由fit_generator或evaluate_generator提供隨機樣本。

另一種可能性,如果這正在對圖像做的是不使用的重新調整= 1 ImageDataGenerator/255爲https://github.com/fchollet/keras/issues/3477

希望有所幫助提及!