NN分类是否输出概率?

时间:2016-12-20 16:42:44

标签: machine-learning neural-network

我刚开始使用张量流与神经网络合作,我对此非常陌生。我训练了我的第一个模型进行了2个类别的分类,我对输出有点好奇。假设我们根据房价是否上涨做出预测,我们会得到像

这样的输出
House A: .99
House B: .75
House C: .55
House D: .40

我可以假设这些输出是概率吗?所以房子B更有可能上涨,而不是House C.或者它只是把它分类为C和B会上升而House D不会。谢谢!

2 个答案:

答案 0 :(得分:2)

不完全是。神经网络将输出您训练的内容的预测。因此,如果你训练它来预测概率,它肯定会输出(预测)概率。但是,如果你根据价格确实上涨的观察进行训练,比如单个输出,如果价格上涨则为1.0,如果价格没有上涨则为0.0,那么输出将是回归值观察给出了输入。这不一定是概率,而是可以被视为模型的置信度。

答案 1 :(得分:1)

是的,每个数字都可以被认为是表示房屋价格上涨的概率。为了进一步澄清,一个房屋的概率估计不影响其他房屋的概率估计,因为它们被视为单独的样本。所以B更有可能不会降低C的可能性。只是B恰好更有可能上升。

分类取决于你的门槛。默认情况下,我认为大多数分类器使用0.5作为阈值,因此在这种情况下,A,B和C被分类为上升,D被分类为下降。