罐装张量流估计器是否可以训练不适合内存的数据集？

Question

我一直在关注Tensorflow中的高级estimator界面，在wide_n_deep tutorial中相当不错。它似乎不允许流输入，我认为我需要一个不适合内存的训练集。

高级API是否支持此功能？我正在读源，我不能说。看起来我可能会编写一个输入函数，它具有生成器而不是数组，但也许代码排除了这一点。

P.S。与this question相关的排序，但我想坚持使用高级API。

Answer 1

使用高级API，您当然可以使用TensorFlow训练不适合内存的数据。只需使用Dataset API即可。你可以搜索： ＆＃34;数据集API支持多种文件格式，因此您可以处理不适合内存的大型数据集。在那个页面。如果要将数据集与估算器一起使用，请搜索＆＃34; input_fn＆＃34;在那个页面上。