如何在keras中为heirachial模型设置非线性损失函数

Question

我试图在keras中建立一个简单的单输出回归模型，在那里我可以对我的预测值有一定的信心。为此，我希望在两个输出上运行自定义丢失函数，如下所示：

def custom_loss(y_true,y_pred,y_error):
    return T.abs_(y_true-y_pred)*y_pred/y_error + T.exp(y_error)

但是我只能输入两个输入y_true和y_pred，那么有没有办法让这个表格有自定义丢失函数？

或者，有没有更好的方法来估计回归的可信度？

Answer 1

在keras中，您可以构造另一个Lambda层来计算错误，然后可以将y_error包含到y_pred中（例如model.outputs = [last_layer.output,Lambda_layer.output]）。