自定义损失函数如何再添加另一个参数呢？

来源：3-5 实战自定义损失函数与DenseLayer回顾

BloodHound_swh

2019-11-16

请问老师，自定义损失函数除了y_pred和y_true这两个参数之外，如何再添加另一个参数呢？要求这个参数在模型训练和预测时所传入的值是不同的。

写回答

1回答

正十七

2019-11-17

通过我们课上讲的那个方法是无法加入新的参数的，一种较快的方法是分别实现你说的对train和eval的损失函数，然后都放到loss里，这样，两个值你都可以看到。

第二种方法复杂一些，也灵活一些，是把损失函数封装到一个自定义层次中去进行计算。具体可以参考https://zhuanlan.zhihu.com/p/74009996

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

回答 2

回答 1

回答 1

回答 1

回答 1