您可以像这样在Keras中使用加权的 MSE
model.fit(sample_weight=weights, loss='mse', ...)
我想使用加权的 RMSE ，但是Keras库没有rmse，我自己写了
def root_mean_squared_error(y_true, y_pred):
return K.sqrt(K.mean(K.square(y_pred - y_true)))
但是如何使用权重？
答案 0 :(得分：1)
从documentation看来，它是自动完成的：
造成自定义损失： 带有签名 loss_fn（y_true，y_pred）的任何可返回损失数组（输入批处理中的一个样本）的可调用对象都可以作为损失传递给 compile（） 。 请注意，此类损失会自动支持样本加权。