Keras flow_from_dataframe错误的数据排序

Question

我正在使用具有flow_from_dataframe的keras数据生成器。为了进行训练，它工作得很好，但是在测试集上使用model.predict_generator时，我发现生成结果的顺序与数据框中“ id”列的顺序不同。

shuffle = False确实使生成器的顺序一致，但是与数据帧的顺序不同。我还尝试了不同的批处理大小，并为predict_generator函数设置了相应的正确步骤。 （例如：batch_Size = 1，steps = len（data））

如何确保以我的数据框“ id”列的相同方式对为测试集预测的标签进行排序？

Answer 1

虽然我还没有找到一种方法来确定生成器生成数据的顺序，但是可以使用generator.filenames属性获得该顺序。