在定制的 keras 训练损失函数中缩减数据

Question

我为我的 LSTM 模型（RMSE 函数）定义了一个自定义损失函数，如下所示：

def RMSE(y_true, y_pred):
        return K.sqrt(K.mean(K.square(y_pred - y_true)))

到目前为止一切都很好，但问题是我将输入数据缩放到 [-1, 1] 范围内，因此报告的损失将与此比例相关联，我希望模型报告我的原始数据范围内的训练损失，例如通过在 y_true 和 y_pred 上以某种方式应用 scaler.inverse_transform 函数，但没有运气......因为它们是张量和scaler.inverse_transform 需要 numpy 数组....

知道如何强制重新缩放数据并以正确的比例报告损失值吗？

Answer 1

scaler.inverse_transform本质上是使用scaler.min_和scaler.scale_参数来转换sklearn.preprocessing.minmaxscaler中的数据。一个例子：

from sklearn.preprocessing import MinMaxScaler
import numpy as np

data = np.array([[-1, 2], [-0.5, 6], [0, 10], [1, 18]])
scaler = MinMaxScaler()
data_trans = scaler.fit_transform(data)
print('transform:\n',data_trans)

data_inverse = (data_trans - scaler.min_)/scaler.scale_
print('inverse transform:\n',data_inverse)

# print
transform:
 [[0.   0.  ]
 [0.25 0.25]
 [0.5  0.5 ]
 [1.   1.  ]]
inverse transform:
 [[-1.   2. ]
 [-0.5  6. ]
 [ 0.  10. ]
 [ 1.  18. ]]

所以你只需要使用它们来实现你在 RMSE 函数中的目标。

def RMSE_inverse(y_true, y_pred):
    y_true = (y_true - K.constant(scaler.min_)) / K.constant(scaler.scale_)
    y_pred = (y_pred - K.constant(scaler.min_)) / K.constant(scaler.scale_)
    return K.sqrt(K.mean(K.square(y_pred - y_true)))

在定制的 keras 训练损失函数中缩减数据

scaling back data in customized keras training loss function

keras

tensorflow

loss-function