Tensorflow中的_Head对象是什么?

时间:2017-04-13 20:45:29

标签: tensorflow

查看DNNLinearCombinedEstimator的文档,我看到第一个参数是一个_Head对象:

  

参数数量:

     

head:一个_Head对象。

我根本找不到关于此的文档。

这是什么?

2 个答案:

答案 0 :(得分:3)

我发现了这个:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/learn/python/learn/estimators/head.py#L53

  

模型的头部/顶部的接口。

     

给定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