假设我有一个已经过训练的Keras模型。当使用predict（）方法时，我想同时获取实例键值和相应的预测（我可以在输入中将键值作为要素/列传递）。
我想知道这样做是否现实吗？
我为此苦了一段时间。我正在使用tf.data.Dataset基础结构，所以我的第一种方法是查看是否可以确保由数据集生成的示例的顺序是确定性的，但这并不是最佳选择，因为它放弃了很多并行处理性能好处，最终无论如何都不是。我最终使用model.predict_on_batch手动处理从数据集中迭代的数据进行处理，而不是将整个数据集馈入model.predict。这样，我就可以从批处理中获取ID，并将其与返回的预测相关联。
我感到惊讶的是，还没有一个现成的解决方案可以解决很多问题。我还没有赶上Estimator界面或自定义训练/预测循环的速度，但希望这个问题在这里变得微不足道。