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

Question

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

Answer 1

正十七 2019-11-17 18:31:45

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

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