代码之家 › 专栏 › 技术社区 › Vaaal88

Tensorflow/Keras自定义损失函数

keras tensorflow

Vaaal88 · 技术社区 · 6 年前

我想在keras中创建一个自定义损失函数。

假设我有yTrue和yPred,它们是真标签和预测标签的张量(nx m)。

让我们将每个示例n(即,yTrue和yPred中的每一行)称为yT和yP。然后我想要一个损失函数,当yT[0]=1时计算(yT yP)^2,否则它将计算(yT[0]-yP[0])^2。

也就是说:对于每个样本,我总是想计算第一个元素的平方误差——但我只想在真标签的第一个元素=1时计算其他元素的平方误差。

如何在自定义损失函数中执行此操作?

到目前为止,我得到的是:

我需要做张量运算。首先我可以计算

Y = (yTrue - yPred)^2

然后我可以定义一个屏蔽矩阵,其中第一列始终为一,其他列为1,这取决于yTrue每行的第一个元素的值。这样我就可以

然后我可以将这个矩阵按元素乘以Y,得到我想要的结果。

但是,如何生成掩蔽矩阵?特别是,如何在tensorflow/keras中实现“如果行的第一个元素为1”的条件?

也许有更好的方法? 谢谢

1 回复 | 直到 6 年前

nuric 6 年前

可以使用条件开关 K.switch 在后端。大致如下:

mse = K.mean(K.square(y_pred - y_true), axis=-1) # standard mse
msep = K.square(y_pred[:,0] - y_true[:,0])
return K.switch(K.equals(y_true[:,0], 1), mse, msep)

编辑以根据样本条件进行处理。

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

Anshuman Sinha · 将tensforflow中的所有nan值转换为零

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

rob me of my freedom · 在不使用Y_pred和Y_actual的tensorflow中编写自定义损失

2 年前

Chris · 如何捕获导入脚本功能的打印

2 年前

Jay Bhie Santos · 剩余GRU:ValueError:输入具有不兼容的形状。收到的形状(5)和(24)

2 年前

Danijel · C中TensorFlow模型文件的路径?

2 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

2 年前

Arya Anish · tensorflow安装中面临的问题

2 年前

so beautiful memory · 如何将我在本地训练的tensorflow模型正确部署到谷歌云人工智能平台?我部署了它,但没有图像返回

2 年前