Python 在keras中实现自定义成本函数,该函数需要整个批量的输出
有关数据集的简要信息:- 这是一个人识别问题(无论两张图像是否相同),我必须编写自己的损失/成本函数(二项式偏差)来实现它。 对于128的批大小,我将有256个图像,因此是一个3 256*256/批矩阵 我的疑问是如何使用keras获得批量大小的输出,以便实现成本函数 我添加了二项式偏差成本函数公式的截图。Python 在keras中实现自定义成本函数,该函数需要整个批量的输出,python,tensorflow,computer-vision,keras,Python,Tensorflow,Computer Vision,Keras,有关数据集的简要信息:- 这是一个人识别问题(无论两张图像是否相同),我必须编写自己的损失/成本函数(二项式偏差)来实现它。 对于128的批大小,我将有256个图像,因此是一个3 256*256/批矩阵 我的疑问是如何使用keras获得批量大小的输出,以便实现成本函数 我添加了二项式偏差成本函数公式的截图。 使用附加的批量大小参数定义自定义损失: def my_loss_template(y_true, y_pred, batch_size): # compute the loss 要
使用附加的
批量大小
参数定义自定义损失:
def my_loss_template(y_true, y_pred, batch_size):
# compute the loss
要使用它
batch_size = 20
my_loss = lambda x, y: my_loss_template(x, y, batch_size)
model.compile(optimizer='adam', loss=my_loss)
谢谢,我会尽力实施并回复你。