老师为何predict的值是取每行里的最大值？

Question

Answer 1

正十七 2019-03-20 22:35:45

因为每一行得到的是样本在各个类别上的概率分布，比如5个类，那么每一个样本得到的是

[0.1, 0.05, 0.05, 0.7, 0.1], 这样取最大值得到的4就是类别id

• 提问者 Liny77 #1

  老师我们预测值只要概率最大的类别吗，其他类别不要了吗？抱歉老师这里不太理解！不懂为什么取最大值作为预测值！

• 提问者 Liny77 #2

  谢谢老师回复老师辛苦了！

• 正十七 回复 提问者 Liny77 #3

  如果需要topN的结果，其他类别就可能会被需要了啊。