【发布时间】:2020-10-12 14:52:53
【问题描述】:
如何在 keras 回归中编写自定义损失函数,其中 MAE 仅针对高于某个阈值的 y_pred 计算。
例如。 y_true = [10 , 14 , 23 , 30 , 5 , 4] ,
y_pred = [8 , 12 , 27 , 38 , 10 , 8]
如何编写自定义损失函数,其中仅计算 y_pred 值大于 20 的 MAE(平均绝对误差),即 y_pred > 20 的 MAE 即 [23,30] 得出 6 [(27-23) + (38 -30)]/2
这个问题出现了,因为我需要只正确预测最高预测范围的模型,这样我就可以只使用那些返回最高预测的数据点,因为其余的较低预测数据对我没有用。 像 -
def custom_loss(y_pred, y_true):
for y_pred > 20:
result =MAE(y_pred , y_true)
return result
【问题讨论】:
标签: python tensorflow keras deep-learning neural-network