神经网络给出元素属于特定类别的概率

时间:2018-08-17 18:26:09

标签: artificial-intelligence

哪种神经网络适合确定某个元素属于特定类别的概率?

我已经看到了确定顶级预测概率的神经网络示例。也就是说,当我上传猫的图片时,神经网络会确定这是一只猫,概率为80%,而这只豹的概率为20%

1 个答案:

答案 0 :(得分:1)

概率是最后阶段转换函数的属性,而不是net类型的属性。您所需要的就是该模型可以通过一些排序合理的评估(例如数字)解决分类问题。尝试搜索对 SoftMax (一种流行的转换)的引用。

关键部分是更改网络预测的报告。例如,输出置信度百分比最高的五个类别(而不只是顶部类别的索引)。这是最终返回值的更改。