tensorflow - Tensorflow 中的 _Head 对象是什么？

Question

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

Args:

head: A _Head object.

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

这是什么？

Answer 1

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

Interface for the head/top of a model.

Given logits (or output of a hidden layer), a Head knows how to compute predictions, loss, default metric and export signature.

继续阅读，它看起来只是一些对象，您可以从中获得模型的预测、损失和更多信息，创建它是为了简化 model_fn()。通常每个模型目标都有一个头部对象。