我想缩放模型输出并重新规范化它以处理类不 balancer 问题 . 例如,如果我有10个标签输出 y_logits
及其softmax y_pred
和之前的 p
,则新输出应为:
y_pred /= prior
y_pred /= sum(y_pred)
问题是张量流中的 softmax_cross_entropy_with_logits
函数采用logits y_logits
而我需要在 y_pred
上进行此缩放 . 任何想法如何在不实施交叉熵损失的情况下做到这一点?
1 回答
对于那些面临同样问题的人,我直接实现CE,因为它的等式表示
-∑ p_i log(q_i)
请查看tutorial .我用来应用先验的实现如下: