python - 使用 DNNRegressor 的损失函数是什么？

Question

我正在使用 DNNRegressor 来训练我的模型。我在文档中搜索此包装器使用的损失函数是什么，但我没有找到。另一方面，有可能改变那个损失函数吗？。

感谢您的建议。

Answer 1

它使用 target_column.py 中定义的 L2 损失(均方误差) :

def regression_target(label_name=None,
                      weight_column_name=None,
                      target_dimension=1):
  """Creates a _TargetColumn for linear regression.
  Args:
    label_name: String, name of the key in label dict. Can be null if label
        is a tensor (single headed models).
    weight_column_name: A string defining feature column name representing
      weights. It is used to down weight or boost examples during training. It
      will be multiplied by the loss of the example.
    target_dimension: dimension of the target for multilabels.
  Returns:
    An instance of _TargetColumn
  """
  return _RegressionTargetColumn(loss_fn=_mean_squared_loss,
                                 label_name=label_name,
                                 weight_column_name=weight_column_name,
                                 target_dimension=target_dimension)

并且目前 API 不支持此处的任何更改。但是，由于它是开源的 - 您始终可以修改构造函数以在内部调用不同的函数，并具有不同的损失。