我正在使用 TensorFlow 处理回归问题，我的损失函数遇到了挑战。我的数据点的结构为三元组 $(Y_i, G_i, X_i)$，其中 $Y_i \in \mathbb{R}$ 代表结果； $G_i \in {0,1}$ 是二进制组标识符； $X_i \in \mathbb{R}^d$ 是一个特征向量。
目标是训练一个神经网络，在给定 $X$ 的情况下预测 $\hat{Y}$，使用自定义损失函数，该函数是两组之间均方误差 (MSE) 的绝对差。
形式上，预测算法的损失函数 $f