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

来源:2-8 神经网络实现(多分类逻辑斯蒂回归模型实现)

Liny77

2019-03-20

图片描述

写回答

1回答

正十七

2019-03-20

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

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

0
3
正十七
回复
Liny77
如果需要topN的结果,其他类别就可能会被需要了啊。
2019-04-09
共3条回复

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程