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

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

扫码关注官方订阅号 20231103003222_72706.jpg" alt="机器学习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|苏州跃动光标网络科技有限公司|


本文地址: https://www.gpxz.com/article/00be0f245f5e03d2b5dd.html
全局中部横幅
全局中部横幅
北京市2026年税后工资个税计算器及周边服务

张大妈工资计算器(hizdm.cn)按照最新的五险一金缴纳比例计算各城市的税后工资收入,帮助您更详细了解五险一金扣税的各比例和金额。

视频

澎湃是植根于中国上海的时政思想类互联网平台,以最活跃的原创新闻与最冷静的思想分析为两翼,是互联网技术创新与新闻价值传承的结合体,致力于问答式新闻与新闻追踪功能的实践,澎湃,澎湃新闻,澎湃新闻网,新闻与思想

江苏密封胶厂家

江苏雅世菲特新材料有限公司是一家比较有名的密封胶生产厂家,专注于密封胶,硅酮胶,一体板专用密封胶,一体板辅材,保温填缝剂,粘接剂,结构胶等产品生产与销售,厂家发货,质量可靠,价格公道,欢迎来电采购批发密封胶,硅酮胶:18352509783.

郑州酒店照片拍摄

郑州元熹文化传播有限公司专业从事VR全景拍摄、720度全景展示、酒店照片拍摄、民宿照片拍摄、商业空间拍摄、酒店客房照片拍摄、提供成熟的三维全景展示解决方案,为众多企业提供了稳定可靠的全景拍摄制作服务,我们在河南省内各县市区均有专业的拍摄和制作团队!

浙江科菲科技股份有限公司

请用一段语句通顺的话来描述您的网站定位,字数不超过200字。

复合面料

昆山市英杰纺织品进出口有限公司是一家专业生产专业户外运动面料,复合面料,阻燃面料,防紫外线面料为一体的公司,您身边的功能性面料及布料专家,公司主要生产功能性面料,复合面料,TPU复合面料,阻燃面料,防紫外线面料,吸湿排汗面料,防水透气面料,箱包面料,滑雪服面料,羽绒服面料为主的企业。

设序科技

闪设CAD是设序科技自主研发的工业AI+CAD软件,基于生成式AI技术和自研的垂直生成模型帮助用户实现3D数模转2D图纸,自动设计3D模型,闪设CAD软件已应用成功于汽车、3C、航空航天、物流、能源等工业场景下的装备、零部件设计、出图

麻花影视

麻花影视(www.weixintools.net)是专门为想要观看视频的用户设计的移动UC、百度、夸克电影播放器。该软件具有最强大的电影和电视播放功能,您可以免费观看所有流行的神马大片电影最新最好看的电影电视剧全集下载就来麻花电影看!

江门和美网络工程有限公司

江门和美网络专业承接工厂防盗工程、工厂监控工程、门禁考勤系统、网络工程、小区道闸系统、LED广告系统安装工程。联系电话:18814157711。公司网站:www.hemeinet.com

全局底部横幅