我可以在不使用要素列的情况下制作自定义tf.estimator吗？

Question

即。我可以直接将具有数字数据的张量输入到自定义估算器的model_fn中吗？

创建一个数字列并执行从键到值的映射似乎对我来说太过分了，我只对图像数据起作用。

Answer 1

是。 features的{​​{1}}参数可以简单地为张量，或者在多个输入的情况下将字符串映射到张量。这也意味着您的model_fn可以简单地返回此类对象。