使用Keras Model.predict_generator时如何关联训练数据ID

Question

问题

我有测试数据，其中每个示例都包含唯一的id以及许多功能。如果我使用Model.predict_generator，则作为传递我作为参数之一的生成器来加载数据，这将成功地将测试数据的特征传递给我的模型并吐出预测。但是，我也想通过我的id属性，并获得与id关联的预测。测试数据太大，无法容纳到内存中。

可行的解决方案

我可以直接从生成器中读取数据，而不是将其提供给Model.predict。但是，这似乎是一种解决方法，而且我觉得很多人也偶然发现了这一点。