Keras flow_from_dataframe错误的数据排序

Posted 2021-05-04

我正在使用keras的数据生成器和flow_from_dataframe。对于训练它工作得很好，但是当在测试集上使用model.predict_generator时，我发现生成结果的排序与我的数据帧中“id”列的排序不同。

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

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

答案

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