我是 Keras 新手,正在尝试实现神经网络机器学习模型。输入张量看起来像 (X1, X2) 和输出 (Y)。注意 X1 和 X2 是相关的。在模型中,只有 X1 将用于训练,但 X1 和 X2 都将传递给损失函数,该损失函数是 X1、X2、y_pred 和 y_true 的函数。下面是损失函数的伪代码。
def customLossFunctionWrapper(input_tensor):
def LossFunction(y_pred, y_true):
loss_value = f(X1, X2, y_pred, y_true)
return loss_value.
我的问题不同于Keras 自定义损失函数:访问当前输入模式 https://stackoverflow.com/questions/46464549/keras-custom-loss-function-accessing-current-input-pattern。我需要在训练模型中屏蔽 input_tensor 的一部分,但在损失函数中使用屏蔽部分。
有人可以给我一些想法吗?
先感谢您。
None
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)