如何在Tensorflow中用罐装估计器实现自定义损失函数?

2024-10-02 20:35:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我对tensorflow很陌生。你知道吗

问题:我怎样才能把一个自定义的损失函数添加到一个像DNNLinearCombinedRegressor这样的封闭估计中?你知道吗

提前感谢您的帮助!你知道吗


Tags: 函数tensorflow损失陌生dnnlinearcombinedregressor
1条回答
网友
1楼 · 发布于 2024-10-02 20:35:08

我没有找到loss_fn参数来传递给DNNLinearCombinedRegressor。但定义任何自定义估计器看起来都很简单:

一个好的参考是code表示DNNLinearCombinedRegressor。你知道吗

我相信你需要以下步骤:

  1. 定义一个定制的loss_fn(你喜欢什么)
  2. 定义或使用上面定义的RegressionHead自定义loss_fnuses。你知道吗
  3. pass自定义head到您自己的Regressor。当然,您可能还需要复制/构建自己的model_fn。你知道吗

相关问题 更多 >