如何在keras注册自定义损失函数?

inkz8wg9  于 2023-04-06  发布在  其他
关注(0)|答案(1)|浏览(135)

我正在尝试使用这个自定义损失函数

# calculate binary cross-entropy loss
    lbce = -(1/N*M) * np.mean(y_actual*np.log(y_pred) + (1-y_actual)*np.log(1-y_pred))
    
    # calculate mean squared loss with normalization
    lmsl = 1/(N*M) * np.mean(np.square(y_pred - y_actual))
    
    # add the two losses together
    loss = lbce + lmsl
    
    return loss

我得到这个错误ValueError:未知损失函数:'vae_loss'。请确保您使用的是keras.utils.custom_object_scope,并且此对象包含在作用域中。
有人知道怎么解吗?

eqzww0vc

eqzww0vc1#

Meh,当你为keras定义自定义损失时,不要使用numpy。你应该使用keras或tensorflow。我建议你使用tensorflow函数。
例如,使用tf.reduce_mean代替np.mean...

相关问题