每个输入具有不同重量的损失函数

时间:2017-03-08 09:08:12

标签: tensorflow keras

我想设计我的自定义丢失函数,它不仅包含两个输入(y_real和y_pred),而且我不知道我是否能够做到这一点以及如何做。

我已经完成了关于损失函数的答案,所有这些都只输入了y_real和y_pred。

但是在我的项目中,真假预测的丢失是多种多样的,例如X1带有(loss_T1,loss_F1),X2带有(loss_T2,loss_F2)。

因此对于我的损失函数,我需要输入四个变量(y_real,y_pred,loss_T,loss_F)。 loss_T,loss_F在X(我的数据的每一行)内部作为一个特征。

feature1 feature2 feature3 Loss_T Loss_F y X1
X2
X3

我有办法吗?

感谢您的出色贡献。

1 个答案:

答案 0 :(得分:1)

较低级别的tensorflow API可以为您服务。例如,Optimizer.optimize_loss采用损失张量,它只是损失函数的输出,并将其最小化。因此,如果你直接使用optimize_loss,你可以用你想要的任何复杂方式建立你的损失函数。