如何将RMSE添加到预制估算器

时间:2019-07-01 22:14:50

标签: tensorflow-estimator

我一直在努力为我的预制估算器模型有效地添加RMSE输出,就像使用keras的火车函数时一样。我一直在考虑使用add_metrics,但是甚至不确定是否可以将其用于预制估算器,如果可以,如何使用?这意味着我需要如何编码metric_fn?

谷歌通过调用预测并将其转换为np.array的方式对我来说需要年龄。

我很高兴收到关于如何进行这项工作的任何想法。

谢谢!

0 个答案:

没有答案