tensorflow - 加权交叉熵张量流

Question

我找不到允许您传入不等于 1 的标签的内置 tensorflow，因此尝试编写自己的：（输入为 [batch_size,labels]）

tf.reduce_mean(tf.reduce_sum(y_true,axis=1) * tf.reduce_logsumexp(y_pred_logits,axis=1)
 - tf.reduce_sum(y_true * y_pred_logits,axis=1))

然而，它似乎没有工作（损失是不同的）。我做错什么了吗？

Answer 1

我认为这是正确的，但我将 Adam 优化器的“epsilon”设置得太低。