机器学习|PyTorch简明教程上篇-人工智能

文章编号:5913 技术教程 2023-11-03 机器学习 PyTorch

扫码关注官方订阅号 机器学习PyTorch简明上篇人工智

前面几篇文章介绍了特征归一化和张量,接下来开始写两篇PyTorch简明教程,主要介绍PyTorch简单实践。

图1

(2)torch.linspace(-1,1,101,requires_grad=False)用于在指定的区间内生成一组等间隔的数值,该函数接受三个参数:起始值、终止值和元素个数,返回一个张量,其中包含了指定个数的等间隔数值;

使用optim.SGD(model.parameters(),lr=0.01,momentum=0.9)可以实现随机梯度下降(StochasticGradientDescent,SGD)优化算法

将训练集通过批量大小拆分,循环100次

(7)接下来是训练函数train,用于训练一个神经网络模型,具体来说,该函数接受以下参数:

(8)train是PyTorch训练过程中常用的方法,其步骤如下:

(9)print("轮次=%d,损失值=%s"%(i1,cost/num_batches))最后打印当前训练的轮次和损失值,上述的代码输出如下:

图2

(3)build_model内部实现:torch.nn.Sequential(torch.nn.linear(input_dim,output_dim,bias=False))用于构建一个包含一个线性层的神经网络模型,模型的输入特征数量为input_dim,输出特征数量为output_dim,且该线性层没有偏置项,其中n_classes=10表示输出10个分类;重写后:(3)build_model内部实现:使用torch.nn.Sequential(torch.nn.Linear(input_dim,output_dim,bias=False))来构建一个包含一个线性层的神经网络模型,该模型的输入特征数量为input_dim,输出特征数量为output_dim,且该线性层没有偏置项。其中n_classes=10表示输出10个分类;

(4)其他的步骤就是定义损失函数,梯度下降优化器,通过batch_size将训练集拆分,循环100次进行train;

使用optim.SGD(model.parameters(),lr=0.01,momentum=0.9)可以实现随机梯度下降(StochasticGradientDescent,SGD)优化算法

(6)在每一轮训练结束后,需要执行predICT函数来进行预测。该函数接受两个参数model(已经训练好的模型)和teX(需要进行预测的数据)。具体步骤如下:

(7)print("Epoch%d,cost=%f,acc=%.2f%%"%(i1,cost/num_batches,100.*np.mean(predY==teY)))最后打印当前训练的轮次,损失值和acc,上述的代码输出如下(执行很快,但是准确率偏低):

图3

(2)print("Epoch%d,cost=%f,acc=%.2f%%"%(i1,cost/num_batches,100.*np.mean(predY==teY)))最后打印当前训练的轮次,损失值和acc,上述的代码输入如下(执行时间比逻辑回归要长,但是准确率要高很多):

以上就是机器学习|PyTorch简明教程上篇的详细内容,更多请关注php中文网其它相关文章!

微信扫码关注PHP中文网服务号

QQ扫码加入技术交流群

Copyright2014-2023AllRightsReserved|苏州跃动光标网络科技有限公司|


本文地址: http://www.gpxz.com/article/00be0f245f5e03d2b5dd.html
全局中部横幅
APP开发

【官网】起翌软件针对不同行业特点厂商及分销商、连锁店、零售商、本地生活服务、社区O2O、批发市场、百货商场、跨境电商等业态提供专业的场景化定制解决方案。主营业务:APP开发、微信定制开发、网站个性化定制、软件开发,为客户打造一站式解决方案。咨询热线:

深圳海蓝机电设备

深圳市海蓝机电设备是一家专业从事三菱自动化20年,主营三菱PLC、三菱变频器、三菱触摸屏、三菱伺服电机、三菱机器人等工业自动化产品。专业售前、售后服务团队拥有丰富的实战经验,能以最快速度满足用户对三菱产品的各种需求。

毕业论文查重

放心测fangxince.cn为论文查重入口,查重结果与学校一致!论文检测系统是学校使用的官方论文检测系统,可供本科论文查重、职称论文检测、PAPEROK、PAPERFREE、硕士论文查重、博士毕业论文查重、维普论文检测、PAPERPASS论文查重,万方论文查重,助您轻松通过高校本硕博毕业论文检测。

江苏华正环保科技有限公司【官网】

江苏华正环保科技有限公司从事工业环保除尘设备、废气治理设备、异味处理设备、污水处理设备、粉尘防爆设备,如需了解相关介绍可联系爆炸隔离阀阻火器厂家.

加热炉

燃气加热炉-开封现代工业炉有限公司是专业从事热处理炉窑设计、制造、现场安装调试的企业,同行业内率先通过3C认证,专业生产各种工业电炉。咨询电话:13837856639(张经理)

哈尔滨中招国联科技有限公司官网

哈尔滨中招国联科技有限公司成立于2015年,主要承载zhaobiao.cn相关的营销业务,并针对采购业主和招标机构提供一系列服务。

全球品牌联盟媒体发布平台

全球品牌联盟媒体发布平台,全网发布各种新闻和软文

陆良本草秀复祛痘连锁

本草秀复专业祛痘(中国)连锁,祛痘祛痘印就到本草秀复。全国连锁专业祛痘机构,专注于解决青春痘、粉刺、痘印、痘坑、黑头、毛孔粗大等问题。本草秀复祛痘热线:18313502141。祛痘哪里好,祛痘去痘印就选本草秀复。

企业服务批发超市

企业一站式服务平台(www.stpp.cc)以“整合行业资源,服务大众企业”为宗旨,致力打造全面、优质、省钱、省心的企业服务平台。

电子工程世界

电子工程世界(EEWorld)是一家专为中国电子工程师和电子设计主管提供电子技术开发应用资讯的网络传媒。其内容服务核心是快速传播半导体集成电路领域电子元器件的最新技术产品,深入挖掘并分享各类电子设备开发经验和电子技术应用知识,提供电子工程师和设计主管所需要的各种工具资料。频道包括:汽车电子,手机便携,数字电视,网络通信,工业控制,测试测量,安防电子,医疗电子,单片机,嵌入式,模拟电子,DSP,电源管理,FPGA,RF无线,光电显示,传感技术,缓冲存储,MEMS,半导体设计制造,LED

海外仓系统就选易仓科技ECCANG

专业的海外仓系统,适用于一件代发、FBA中转、退换标等多种业务管理,订单处理效率提升45%,经营成本降低35%,全球500多家海外仓信赖之选。

永发资源网

永发资源网是生活帮助资讯平台,包含便民信息,教育,交通,旅游等实用民生帮助信息。


全局底部横幅