首页 > 解决方案 > 如何使用部分输入进行训练,但在 Keras 中使用损失函数

问题描述

我是 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 自定义损失函数不同:Accessing current input pattern。我需要在训练模型中屏蔽 input_tensor 的一部分,但在损失函数中使用屏蔽部分。谁能给我一些想法?先感谢您。

标签: pythonmachine-learningkeras

解决方案


推荐阅读