答案 0 :(得分:3)
模型的头部/顶部的接口。
给定logits(或隐藏层的输出),Head知道如何 计算预测,损失,默认度量和导出签名。
阅读,它看起来只是一些对象,您可以从中获得预测,损失以及更多模型,以简化model_fn()。而且每个模型目标通常只有一个头部对象。
答案 1 :(得分:0)
看起来它会在将来的Tensorflow的正式版本中包含在新的Head API中,现在它位于contrib
模块中。您可以在这里阅读有关设计的更多信息:https://github.com/ewilderj/tensorflow/blob/aea71e8ab41dc18478de019a5b82a66ae76f6e81/rfc/estimator-head.md
此外,其中一位创作者还提供了一个小演示:https://youtu.be/4oNdaQk0Qv4?t=11m30s