版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/w_q_q2017/article/details/79183037
Softmax回归模型是logistic回归模型在多分类问题上的推广,在多分类问题中,待分类的类别数量大于2,且类别之间互斥。
Softmax公式:
通常情况下softmax会被用在网络中的最后一层,用来进行最后的分类和归一化。
通常有2种结构:
第一种
#可以计算给出每个样本对应的损失函数值
layer {
name: "loss"
type: "SoftmaxWithLoss"
bottom: "ip2"
bottom: "label"
top: "loss"
}
第二种
#输出为每个类别的概率值
layers {
name: "prob"
type: “Softmax"
bottom: " ip2"
top: "prob"
}