1. 首页
  2. 人工智能
  3. 深度学习
  4. pytorch实现softmax多重分类

pytorch实现softmax多重分类

上传者: 2020-08-20 08:04:25上传 IPYNB文件 6.09KB 热度 34次
代码清晰的说明使用pytorch实现多重分类, 有十个分类标签,用到的思路就是贝叶斯朴素分类器的思想,多分类问题参照BCELoss的函数表达式可变成loss(yhat, y) = -y*log(yhat),Torch.nn.EntropyLoss()交叉熵损失:包含softmax,log处理loss计算和非线性激活
下载地址
用户评论