什麼是Tensorflow中的_Head對象？

Question

望着文檔的DNNLinearCombinedEstimator，我看到的第一個參數是一個_Head對象：

參數數量：

頭：一個_Head對象。

我找不到關於此的文檔。

這是什麼？

Answer 1

我發現這一點：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/learn/python/learn/estimators/head.py#L53

接口模型的頭部/頂。

鑑於logits（或隱藏層的輸出），頭部知道如何計算預測，損失，默認度量和導出簽名。

讀起來，它看起來只是一些對象，您可以從中獲取模型的預測，損失和更多信息，以簡化model_fn（）。而且，每個模型目標通常都有一個頭對象。