python - 损失函数 Keras out_dim > 1

Question

我有一个训练数据:

而且，我在 Keras 中有一个模型，具有多个输出维度。我想预测 A、B 和 C:

model = Sequential()
model.add(GRU(32, input_shape=(train_X.shape[1], train_X.shape[2])))
model.add(Dense(3))
model.compile(loss='mean_squared_error', optimizer='adam')

但我想要 A 中的最小 mean_squared_error，即只想考虑 A 作为损失函数。

我能做什么？

Answer 1

您可以定义自定义损失函数，并仅根据 A 的值计算 mean_squared_error() 损失:

from keras import losses

def loss_A(y_true, y_pred):
    return losses.mean_squared_error(y_true[:,0], y_pred[:,0])

#...
model.compile(loss=loss_A, optimizer='adam')