加权交叉熵张量流

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”设置得太低了。