如何在 Tensorflow 中使用固定估计器实现自定义损失函数？

Question

我是张量流的新手。

问题：如何向 DNNLinearCombinedRegressor 这样的固定估计器添加自定义损失函数？

提前感谢您的帮助！

Answer 1

我没有找到要传递给DNNLinearCombinedRegressor 的loss_fn 参数。但是定义任何自定义估算器看起来很简单：

一个很好的参考是code for DNNLinearCombinedRegressor。

我相信您需要以下步骤：

1. 定义一个自定义的loss_fn（随你喜欢）
2. 定义或使用RegressionHead 即uses 上面定义的自定义loss_fn。
3. pass 自定义 head 到您自己的 Regressor。当然，您可能还需要复制/构建自己的 model_fn。