如何在 Keras 中使用部分输入进行训练,其余部分用于损失函数

2024-05-10

我是 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(使用前将#替换为@)

如何在 Keras 中使用部分输入进行训练,其余部分用于损失函数 的相关文章

随机推荐