python - Keras:平均不同的损失函数

Question

我正在使用带有 TensorFlow 后端的 Keras，并且想要定义如下所示的自定义损失函数:

• 它计算 y_true 和 y_pred 的前 2 个条目之间的欧氏距离
• 它计算其余条目之间的绝对误差

然后取平均值。

我这样做:

def custom_objective(y_true, y_pred):
    y_true = backend.get_value(y_true)
    y_pred = backend.get_value(y_pred)

    a = np.sqrt(np.mean(np.square(y_pred[:2] - y_true[:2]), axis=-1))
    b = np.sum(np.abs(y_true[2:] - y_pred[2:]))
    return (a + b) / 5

编译模型时出现 InvalidArgumentError:

model.compile(loss=custom_objective, optimizer='adam')

Answer 1

您正在 Keras 张量上使用 NumPy，不幸的是，这是一个致命的组合。您正在寻找的是以下内容:

def custom_objective(y_true, y_pred):
  a = K.sqrt(K.mean(K.square(y_pred[:2] - y_true[:2]), axis=-1))
  b = K.sum(K.abs(y_true[2:] - y_pred[2:]))
  return (a + b) / 5 # these operators work on tensors