相关疑难解决方法(0)

Keras自定义丢失函数,用于传递y_true和y_pred以外的参数

我正在编写一个keras自定义丢失函数,其中我想传递给下面的函数:y_true,y_pred(这两个将自动传递),模型内层的权重和常量.

如下所示:

def Custom_loss(y_true, y_pred, layer_weights, val = 0.01):
    loss = mse(y_true, y_pred)
    loss += K.sum(val, K.abs(K.sum(K.square(layer_weights), axis=1)))
    return loss
Run Code Online (Sandbox Code Playgroud)

但上面的实现给了我错误.我怎样才能在keras中实现这一目标?

keras

10
推荐指数
1
解决办法
8526
查看次数

标签 统计

keras ×1