keras中的自定义RMSPE损失函数

2024-10-17 06:32:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在keras中定义我自己的损失函数,即均方根百分比误差。RMSPE的定义是:
equation
我已经将我的损失函数定义为:
from keras import backend as K def rmspe(y_true, y_pred): sum = K.sqrt(K.mean(K.square( (y_true - y_pred) / K.clip(K.abs(y_true),K.epsilon(),None) ), axis=-1) ) return sum*100.
,但是在几次迭代之后,它将损失值作为nan。有人能指出我做错了什么吗。谢谢


Tags: 函数fromimportbackendtrue定义askeras