【发布时间】:2018-01-31 18:25:32
【问题描述】:
我有一个用于多标签分类问题的 CNN,作为损失函数,我使用 tf.nn.sigmoid_cross_entropy_with_logits。
从cross entropy equation 我希望输出是每个类的概率,但我得到的是 (-∞, ∞) 中的浮点数。
经过一番谷歌搜索后,我发现由于一些内部归一化操作,每行 logits 在被输入方程之前都可以解释为概率。
我对如何实际输出后验概率而不是浮点数以绘制 ROC 感到困惑。
【问题讨论】:
标签: machine-learning tensorflow cross-entropy