机器学习|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|苏州跃动光标网络科技有限公司|

全局中部横幅
Spark

腾讯游戏成立于2003年,是全球领先的游戏研发和运营商。作为“超级数字场景”理念的倡导者和实践者,腾讯游戏致力于为用户创造高品质数字生活体验,为产业和社会发展创造更多建设性的价值。

北京办公家具厂家

北京黎明办公家具,办公家具十佳品牌,专业办公家具设计,全品类家具生产基地,欢迎来厂考察,免费上门设计!

测速网

测速网(SpeedTest.cn)提供网速测试,网络质量测试,宽带测速,Wi-Fi测速,5G测速,IPv6测速,带宽检测,路由器测速,网关测速,宽带提速,宽带升级,网络加速,内网测速,专网测速,视频测试,游戏测速,直播测速,网络诊断,蹭网检测,物联网监测,网站监测,API监测,Ping测试,路由测试等专业服务,拥有国内外大量高性能测试点,覆盖电信,移动,联通,网通,广电,长城宽带,鹏博士等运营商,Wi-Fi7,Wi-Fi6,FTTR,全屋Wi-Fi。

四川太阳能工程,成都太阳能热水器,太阳能发电,空气能热水器,四川空气能热水工程,四川杰科太阳能公司欢迎你

四川杰科新能源公司专业从事太阳能发电的设计和安装,太阳能光伏组件及配套产品的销售,太阳能热水,空气能热水和各种锅炉热水及地暖工程的设计和安装,欢迎您的光临或来电洽谈。

汽配人

汽配人网(www.qipeiren.com)是国内最具活力的汽车及配件专业网站,涵盖东风、解放、重汽、陕汽、北汽等品牌汽车及配件信息,为汽车汽配行业人士提供即时供求商机、企业信息、行业动态、行业展会、产品参数、汽车及配件价格、维修和保养等

洛阳环氧地坪施工

洛阳骏卡建材科技有限公司提供环氧地坪施工工程,包含无震动防滑车道,密封固化剂地坪,自流平地坪,跑道球场地坪,幼儿园地坪,彩色路面地坪等,及特种工业地坪施工和旧地坪的改造等,施工热线:18538483535

鹰潭新闻网

鹰潭日报社官方新闻网站。

茂名招聘网

茂名招聘网,茂名人才网提供茂名人才招聘网求职信息及茂名人才网招聘信息.为茂名人才网络求职,招聘,提供贴心服务.茂名招聘网是茂名市诚信人力资源服务机构.茂名求职网络招聘,就上茂名招聘网

湖南中钢智能制造集团股份有限公司

湖南中钢智能制造集团股份有限公司(简称:湖南中钢)创建于2001年,总部位于湖南长沙,湖南中钢是一家专业从事钢材贸易、钢材深加工等业务的综合型企业。

全局底部横幅