最小化和最大化损失



我想训练一个自动编码器,使某些观测值的重建误差较低,而另一些观测值的重构误差较高。

from keras.model import Sequential
from keras.layers import Dense
import keras.backend as K
def l1Loss(y_true, y_pred):
return K.mean(K.abs(y_true - y_pred))
model = Sequential()
model.add(Dense(5, input_dim=10, activation='relu'))
model.add(Dense(10, activation='sigmoid'))
model.compile(optimizer='adam', loss=l1Loss)
for i in range(1000):
model.train_on_batch(x_good, x_good) # minimize on low
model.train_on_batch(x_bad, x_bad, ???) # need to maximize this part, so that mse(x_bad, x_bad_reconstructed is high)

我看到了用sample_weight=-np.ones(batch_size)替换???的一些内容,但我不知道这是否适合我的目标。

如果将样本权重设置为负数,那么最小化它实际上会导致其绝对值的最大化。

相关内容

  • 没有找到相关文章

最新更新