【发布时间】:2021-02-18 21:25:21
【问题描述】:
我正在尝试为 keras NN 模型编写自定义损失函数,但似乎损失函数输出了错误的值。我的损失函数是
def tangle_loss3(input_tensor):
def custom_loss(y_true, y_pred):
true_diff = y_true - input_tensor
pred_diff = y_pred - input_tensor
normalized_diff = K.abs(tf.math.divide(pred_diff, true_diff))
normalized_diff = tf.reduce_mean(normalized_diff)
return normalized_diff
return custom_loss
然后我在这个简单的前馈网络中使用它:
input_layer = Input(shape=(384,), name='input')
hl_1 = Dense(64, activation='elu', name='hl_1')(input_layer)
hl_2 = Dense(32, activation='elu', name='hl_2')(hl_1)
hl_3 = Dense(32, activation='elu', name='hl_3')(hl_2)
output_layer = Dense(384, activation=None, name='output')(hl_3)
optimizer = tf.keras.optimizers.Adam(learning_rate=0.001)
model = tf.keras.models.Model(input_layer, output_layer)
model.compile(loss=tangle_loss3(input_layer), optimizer=optimizer)
然后为了测试损失函数是否有效,我创建了一个随机输入和目标向量,并按照我的预期进行了 numpy 计算,但这似乎与 keras 的结果不匹配。
X = np.random.rand(1, 384)
y = np.random.rand(1, 384)
np.mean(np.abs((model.predict(X) - X)/(y - X)))
# returns some number
model.test_on_batch(X, y)
# always returns 0.0
为什么我的损失函数总是返回零?这些答案应该匹配吗?
【问题讨论】:
标签: python tensorflow keras