Keras中的自定义损失函数

Keras中的自定义损失函数,第1张

Keras中的自定义损失函数

您所要做的就是为此定义一个函数,使用keras后端函数进行计算。该函数必须采用真实值和模型预测值。

现在,由于我不确定您的函数中的g,q,x和y是什么,因此我将在此处创建一个基本示例,而不关心它的含义或它是否是实际有用的函数:

import keras.backend as Kdef customLoss(yTrue,yPred):    return K.sum(K.log(yTrue) - K.log(yPred))

可以在此处查看所有后端功能:https : //keras.io/backend/#backend-functions

之后,使用该函数而不是常规函数来编译模型:

model.compile(loss=customLoss, optimizer = .....)


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5674608.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存