【刘二老师】pytorch深度学习笔记【09多分类问题】-北京尧图网络

【刘二老师】pytorch深度学习笔记【09多分类问题】一、概念多分类问题会用到 softmax 分类器。十个输出之间要互相抑制要有竞争性输出的本质是个分布。要求输出是个分布的条件1每一个输出值大于零 2加一起的概率和为1因此在处理多分类问题时前面这些层还用sigmoid层来处理最后一层用softmax层因为softmax可以输出一个分布。二、Softmax层一Softmax 公式因为用了指数满足了输出每项都大于零。又除以了总和所以满足输出加一起和为1。二Softmax层实例架构讲解首先0.20.1-0.1分别是经过了线性变换的输出值。对这三个输出值取e的指数运算得到1.221.110.90。sum求和再divide除以他们的和得到0.380.340.28它们加一起为1。整个绿框就表示softmax函数。三softmax得到分布后损失函数怎么做Loss 【负的原始标签】× log预测输出Y^------- 与二分类的交叉熵形式相似。红色框范围为NLLLoss实现该Loss的计算过程的代码。pytorch中的交叉熵损失整个红框范围都是交叉熵损失。使用交叉熵损失时神经网络的最后一层不要做激活因为激活部分包含在交叉熵损失模块中了。y 需要是 LongTensor长整型张量。交叉熵损失示例代码第一组预测预测值类别分别为201看[ ]中哪个位置数字最大就是哪个类别与 Y一致我们就可以初步判断第一组的损失函数较小。第二组预测022 不太一致损失较大。三、代码及详细解析importtorchfromtorchvisionimporttransformsfromtorchvisionimportdatasetsfromtorch.utils.dataimportDataLoaderimporttorch.nn.functionalasFimporttorch.optimasoptim# prepare datasetbatch_size64transformtransforms.Compose([transforms.ToTensor(),transforms.Normalize((0.1307,),(0.3081,))])# 归一化,均值和方差train_datasetdatasets.MNIST(root../dataset/mnist/,trainTrue,downloadTrue,transformtransform)train_loaderDataLoader(train_dataset,shuffleTrue,batch_sizebatch_size)test_datasetdatasets.MNIST(root../dataset/mnist/,trainFalse,downloadTrue,transformtransform)test_loaderDataLoader(test_dataset,shuffleFalse,batch_sizebatch_size)# design model using classclassNet(torch.nn.Module):def__init__(self):super(Net,self).__init__()self.l1torch.nn.Linear(784,512)self.l2torch.nn.Linear(512,256)self.l3torch.nn.Linear(256,128)self.l4torch.nn.Linear(128,64)self.l5torch.nn.Linear(64,10)defforward(self,x):xx.view(-1,784)# -1其实就是自动获取mini_batchxF.relu(self.l1(x))xF.relu(self.l2(x))xF.relu(self.l3(x))xF.relu(self.l4(x))returnself.l5(x)# 最后一层不做激活不进行非线性变换modelNet()# construct loss and optimizercriteriontorch.nn.CrossEntropyLoss()optimizeroptim.SGD(model.parameters(),lr0.01,momentum0.5)#冲量值设为0.5优化训练过程。# training cycle forward, backward, updatedeftrain(epoch):running_loss0.0forbatch_idx,datainenumerate(train_loader,0):# 获得一个批次的数据和标签inputs,targetdata optimizer.zero_grad()# 获得模型预测结果(64, 10)outputsmodel(inputs)# 交叉熵代价函数outputs(64,10),target64losscriterion(outputs,target)loss.backward()optimizer.step()running_lossloss.item()ifbatch_idx%300299:print([%d, %5d] loss: %.3f%(epoch1,batch_idx1,running_loss/300))running_loss0.0deftest():correct0total0withtorch.no_grad():#表示下面的代码不会计算梯度fordataintest_loader:images,labelsdata outputsmodel(images)#输出每一个样本是个矩阵矩阵里每一行最大值的下标拿出来对应的就是它的分类。_,predictedtorch.max(outputs.data,dim1)# dim 1 列是第0个维度行是第1个维度totallabels.size(0)#labels是一个N×1的矩阵N1size0就代表取N的值。correct(predictedlabels).sum().item()# 推测predicted与label之间的比较真就是1假就是0。print(accuracy on test set: %d %% %(100*correct/total))if__name____main__:#封装进if函数forepochinrange(10):train(epoch)test()一importtorchfromtorchvisionimporttransformsfromtorchvisionimportdatasetsfromtorch.utils.dataimportDataLoaderimporttorch.nn.functionalasFimporttorch.optimasoptimtransforms针对图像做各种各样处理对数据进行原始处理的工具。torch.nn.functional全连接层的激活不再像之前用sigmoid而是用更流行的r elu( ) 作为激活函数。torch.optim优化器包二batch_size64transformtransforms.Compose([transforms.ToTensor(),transforms.Normalize((0.1307,),(0.3081,))])# 归一化,均值和方差train_datasetdatasets.MNIST(root../dataset/mnist/,trainTrue,downloadTrue,transformtransform)train_loaderDataLoader(train_dataset,shuffleTrue,batch_sizebatch_size)test_datasetdatasets.MNIST(root../dataset/mnist/,trainFalse,downloadTrue,transformtransform)test_loaderDataLoader(test_dataset,shuffleFalse,batch_sizebatch_size)先做一个batch_size因为后面有datasets 和 dataloader。除了多了个transform剩下模块与上节课的一样。使用transform模块的原因下面的过程要通过transform的ToTensor来实现。神经网络在处理时希望输入数值比较小最好在-11之间最好能遵从正态分布。所以要把原始的{0……255}转变为图像张量取值为 [01]也是把单通道转变为多通道。C×W×H1×28×28Normalize归一化Nomalize均值标准差满足N01正态分布给神经网络进行训练。三classNet(torch.nn.Module):def__init__(self):super(Net,self).__init__()self.l1torch.nn.Linear(784,512)self.l2torch.nn.Linear(512,256)self.l3torch.nn.Linear(256,128)self.l4torch.nn.Linear(128,64)self.l5torch.nn.Linear(64,10)defforward(self,x):xx.view(-1,784)# -1其实就是自动获取mini_batchxF.relu(self.l1(x))xF.relu(self.l2(x))xF.relu(self.l3(x))xF.relu(self.l4(x))returnself.l5(x)# 最后一层不做激活不进行非线性变换modelNet()全连接网络要求输入是矩阵所以要把12828三阶的张量变成一阶的向量。如何把张量变成向量把图像的每一行拼起来构成一串就做出了向量所以每一行需要28×28784个元素。把张量变为向量的代码x x.view(-1, 784)view函数改变张量的形状变成二阶张量也就是矩阵。该矩阵有784列-1是占位符代表N这个维度N的大小交给 PyTorch 自动计算。【一批多少个图片样本每张图片784个像素】Linear输入输出relu 对每一层算出的结果进行激活最后一层 self.15(x) 不激活直接接到后面的softmax。四训练把一轮循环封装到函数deftrain(epoch):running_loss0.0forbatch_idx,datainenumerate(train_loader,0):# 获得一个批次的数据和标签inputs,targetdata optimizer.zero_grad()# 获得模型预测结果(64, 10)outputsmodel(inputs)# 交叉熵代价函数outputs(64,10),target64losscriterion(outputs,target)loss.backward()optimizer.step()running_lossloss.item()#取item把值拿出来否则要构建计算图。ifbatch_idx%300299:print([%d, %5d] loss: %.3f%(epoch1,batch_idx1,running_loss/300))running_loss0.0X存到 inputY 存到 target每300轮输出一次 running_loss五deftest():correct0total0withtorch.no_grad():#表示下面的代码不会计算梯度fordataintest_loader:images,labelsdata outputsmodel(images)#输出每一个样本是个矩阵矩阵里每一行最大值的下标拿出来对应的就是它的分类。_,predictedtorch.max(outputs.data,dim1)# dim 1 列是第0个维度行是第1个维度totallabels.size(0)#labels是一个N×1的矩阵N1size0就代表取N的值。correct(predictedlabels).sum().item()# 推测predicted与label之间的比较真就是1假就是0。print(accuracy on test set: %d %% %(100*correct/total))max函数predicted函数

【刘二老师】pytorch深度学习笔记【09多分类问题】

相关推荐

RISC-V IDE MounRiver Studio开发实战：TWEN32V RGB

将本地代码上传到gitee仓库

大模型小白入门指南：收藏这份医疗垂直大模型学习资料，从原理到应用全解析！

更多资讯

JSP+MySQL图书购物系统源码解析：MVC分层与部署避坑指南

智能体工程化实战：从Demo到业务级落地的完整路径

自治系统与元进化闭环的工程化落地方法：从概念到可运行架构

Jev模型接入Codex完整指南：从密钥申请到多文件重构实测

Agent记忆海关：用AST扫描与双池隔离根治记忆污染

JavaWeb问卷调查系统全解析：从部署到改造的实战指南

今日更新

我发现了一个新思路：用 Remotion + Claude Code 像写代码一样自动化生成短视频

Windows下 Codex 中 Chrome 和 Computer Use 插件不可用问题排查及解决参考方式：TaoToken 统一 Key 配置与验证

2026 大模型集体涨价：用 Python 做企业 Token 成本测算与选型避坑（附配置）

本周精选

如何彻底离线使用 Laya-CoreML：Hugging Face 下载、缓存机制与 local_files_only 详解

基于1Panel 部署MySQL MCP 服务：TaoToken 统一 Key 接入与 config.toml 配置实战

崇州园区营销网站建设避坑指南：5大注意事项保备案安全

本月热门

我发现了一个新思路：用 Remotion + Claude Code 像写代码一样自动化生成短视频

Windows下 Codex 中 Chrome 和 Computer Use 插件不可用问题排查及解决参考方式：TaoToken 统一 Key 配置与验证

2026 大模型集体涨价：用 Python 做企业 Token 成本测算与选型避坑（附配置）

读完文章，想聊聊您的网站？