什么是DNNRegressor使用的损失函数?

我在使用DNNRegressor训练我的模型。我查看了文档,试图找出这个封装器使用的损失函数,但没有找到。另一方面,是否可以更改这个损失函数?

感谢您的建议。


回答:

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

def regression_target(label_name=None,                      weight_column_name=None,                      target_dimension=1):  """创建用于线性回归的_TargetColumn。  参数:    label_name: 字符串,标签字典中键的名称。如果标签是张量(单头模型),可以为null。    weight_column_name: 定义表示权重的特征列名称的字符串。在训练期间用于降低或提升示例的权重。它将乘以示例的损失。    target_dimension: 多标签的目标维度。  返回:    _TargetColumn的一个实例  """  return _RegressionTargetColumn(loss_fn=_mean_squared_loss,                                 label_name=label_name,                                 weight_column_name=weight_column_name,                                 target_dimension=target_dimension)

目前API不支持在这里进行任何更改。然而,由于它是开源的,您始终可以修改构造函数以在内部调用不同的函数,使用不同的损失函数。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注