您所要做的就是为此定义一个函数,使用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 = .....)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)