Python tf.estimator.PoissonRegressionHead用法及代码示例

使用 tf.nn.log_poisson_loss 为泊松回归创建 Head。

继承自：RegressionHead，Head

用法

tf.estimator.PoissonRegressionHead(
    label_dimension=1, weight_column=None,
    loss_reduction=tf.losses.Reduction.SUM_OVER_BATCH_SIZE, compute_full_loss=True,
    name=None
)

参数

weight_column 由 tf.feature_column.numeric_column 创建的字符串或 NumericColumn 定义表示权重的特征列。它用于在训练期间减轻重量或增加示例。它将乘以示例的损失。
label_dimension 每个示例的回归标签数。这是标签 Tensor 的最后一个维度的大小(通常，它的形状为 [batch_size, label_dimension] )。
loss_reduction tf.losses.Reduction 之一，除了 NONE 。决定如何减少批次和标签维度的训练损失。默认为 SUM_OVER_BATCH_SIZE ，即损失的加权总和除以 batch size * label_dimension 。
compute_full_loss 是否在计算泊松损失时包括常数log(z!) 项。有关完整文档，请参阅tf.nn.log_poisson_loss。
name 头的名字。如果提供，摘要和指标键将以 "/" + name 为后缀。创建操作时也用作name_scope。

属性

logits_dimension 有关详细信息，请参阅base_head.Head。
loss_reduction 有关详细信息，请参阅base_head.Head。
name 有关详细信息，请参阅base_head.Head。

损失是所有输入维度的加权和。也就是说，如果输入标签的形状为 [batch_size, label_dimension] ，则损失是 batch_size 和 label_dimension 的加权和。

头部期望 logits 形状为 [D0, D1, ... DN, label_dimension] 。在许多应用程序中，形状是 [batch_size, label_dimension] 。

labels 形状必须匹配 logits ，即 [D0, D1, ... DN, label_dimension] 。如果 label_dimension=1 ，形状 [D0, D1, ... DN] 也受支持。

如果指定了 weight_column，则权重的形状必须为 [D0, D1, ... DN] , [D0, D1, ... DN, 1] 或 [D0, D1, ... DN, label_dimension] 。

这是作为广义线性模型实现的，请参阅 https://en.wikipedia.org/wiki/Generalized_linear_model

头部可以与罐装估计器一起使用。例子：

my_head = tf.estimator.PoissonRegressionHead()
my_estimator = tf.estimator.DNNEstimator(
    head=my_head,
    hidden_units=...,
    feature_columns=...)

它也可以与自定义 model_fn 一起使用。例子：

def _my_model_fn(features, labels, mode):
  my_head = tf.estimator.PoissonRegressionHead()
  logits = tf.keras.Model(...)(features)

  return my_head.create_estimator_spec(
      features=features,
      mode=mode,
      labels=labels,
      optimizer=tf.keras.optimizers.Adagrad(lr=0.1),
      logits=logits)

my_estimator = tf.estimator.Estimator(model_fn=_my_model_fn)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.estimator.PoissonRegressionHead。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。