以Tensorflow Estimator为食

Question

我正在尝试使用多个层在Tensorflow中执行复杂的函数逼近。该函数将使用大量生成的数据进行训练，因此我希望能够在运行时生成数据，原因很简单，因为必须生成大量数据。我决定尝试使用带有ModelFnOps的Estimator，但我正处于编写训练循环的位置，而且我似乎无法找到有关使用eval(feed_dict=my_feed_dict)之类的内容的文档{{{ 3}}。到目前为止我唯一发现的就是在Estimator上调用fit()，但这需要调用整个数据集（除非我误解了该函数的用途）。有没有办法在循环中输入单个示例或批次来训练Estimator？

Answer 1

您可以通过input function提供数据。此输入函数是一个传递给估计器（或eval / train / predict方法）的第一类函数。

您还可以使用dataset API创建数据馈送器和迭代器，并在输入函数中返回馈线操作。