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

时间:2019-05-27 08:25:32

标签: python tensorflow

我是tensorflow的新手。

问题:如何向自定义损失函数(如DNNLinearCombinedRegressor)添加固定估计器?

提前感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

我没有找到要传递给DNNLinearCombinedRegressor的{​​{1}}参数。但是,定义任何自定义估算器看起来很简单:

DNNLinearCombinedRegressor的{​​{3}}是很好的参考。

我相信您需要执行以下步骤:

  1. 定义自定义loss_fn(随您喜欢)
  2. 定义或使用code上面定义的自定义RegressionHead的{​​{1}}。
  3. uses自定义loss_fn到您自己的head。当然,您可能还需要复制/构建自己的Regressor