pytorch-softmax回归

1、softmax回归模型

softmax回归模型和线性回归一样将输入特征和权重做线性叠加,softmax回归的输出个数等于标签里的类别数。例如在一个图片分类的问题中,输入的图像为2x2像素图片,分别记为x1,x2,x3,x4。对应数据集的真实标签为狗,猫,鸡,记为o1,o2,o3。
Alt
Alt
softmax回归同线性回归一样,也是一个单层神经网络。输出层也是一个全连接层。

2、softmax运算

问题:1.输出层输出的范围不确定。2.很难度量其与真实值之间的误差
解决:softmax运算——将输出值变化成值为正且和为1的概率分布
Alt
softmax回归对样本i分类的矢量计算表达式为:
Alt

3、交叉熵损失函数(cross entropy)——用于分类模型

在这里插入图片描述

4、总结

1、softmax回归适用于多分类问题,它使用softmax运算输出类别的概率分布
2、softmax回归是一个单层神经网络,输出个数等于分类问题的类别个数
3、交叉熵适合衡量两个概率分布的差异

5、为什么pytorch 框架内没有softmax module?

Alt

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值