运行estimator.predict时如何使用钩子获取feed_dict

Question

我正在使用Tensorflow估计器训练模型并尝试预测一些值，然后将预测结果与输入数据一起保存，以便用户可以从一张表中检查预测结果 看看是否正确。

我尝试使用下面的代码，但是feed_dict输出None，我称之为使用dataset对象的预测。

class InsertPredictResultHook(tf.estimator.SessionRunHook):
    def after_run(self, run_context, run_values):
        print(run_context.original_args.fetches)
        print(run_context.original_args.feed_dict)
        print(run_values)

predictions = classifier.predict(input_fn=mydataset,
    hooks = [InsertPredictResultHook()])

如何获取输入张量以及预测结果，以便为用户保存这些数据？

我正在使用tensorflow 2.0 alpha0