| |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
-> 人工智能 -> 3. Pytorch 线性回归实现 -> 正文阅读 |
|
[人工智能]3. Pytorch 线性回归实现 |
作者:recommend-item-box type_blog clearfix |
1. 生成数据集? ? ? ? 我们构造一个简单的人工训练数据集,训练的赝本数为1000, 特征数为2, 我们使用线性回归模型真是权重和偏差b = 4.2以及一个随机噪声来生成标签。 ? ? ? ? 其中噪声服从均值为0,标准差为0.01的正态分布。噪声代表了数据集中无意义IDE干扰。 ? ? ? ? 下面我们生成数据集,其中,features是训练数据特征,labels是标签。 ????????
2. 读取数据? ? ? ? ?Pytorch提供data包来读取数据,由于data常用来做变量,我们将导入的data模块用Data代替,在每次迭代中,我们随机读取包含10个样本的小批量。
3. 定义模型? ? ? ? 首先导入torch.nn模块,实际上,“nn”是neural networks(神经网络)的缩写。该模块定义了大量神经网络的层,nn的核心结构是Module,它是一个抽象概念,既可以表示神经网络的某个层,也可以表示包含多层的神经网络。在实际应用中,往往集成nn.Module,撰写自己的网络/层。 ? ? ? ? 下面先看看如何用nn.Module实现一个线性回归模型。
????????输出:
????????此外,我们还可以用nn.Sequential来更加方便的搭建网络,Sequential是一个有序容器,网络层按照在传入Sequential的序列一次被添加到计算图中。
????????输出:
? ? ? ? 可以通过net.parameters()来查看模型所有的可学习参数,此函数返回一个生成器。
? ? ? ? 输出:
【注意:线性回归层又叫做全连接层】 4. 初始化模型参数? ? ? ? 在使用模型之前,需要对模型参数做初始化操作,如线性回归中的权重和偏差。Pytorch在init模块中提供了多种参数初始化的方法。 ? ? ? ? 我们通过init.normal_将权重参数每个元素初始化为随机采样于均值为0,标准差为0.01的正态分布,偏差初始化为0.
5. 定义损失函数? ? ? ? Pytorch在nn模块中定义了各种损失函数,这些损失函数可以看做一种特殊的层,Pytorch也将这些损失函数实现为nn.Module的子类。 ? ? ? ? 我们现在使用它提供的均方差损失函数。
6. 定义优化器6.1 优化器常用方法? ? ? ? torch.optim模块提供了很多常用的优化算法,比如SGD, Adam, RMSProp等。 ? ? ? ? 我么采用学习率为0.03的小批量随机梯度下降(SGD)为优化器
? ? ? ? 输出:
6.2 我们可以为不同网络设置不同的学习率,比如:
6.2 改变学习率? ? ? ? 改变学习率的方法有两种,一种是修改optimizer.param_groups中对应的学习率;一种是更为推荐的做法---新建优化器,由于optimizer十分轻量级,重新构建开销很小。【注意:但是第二种方法对于使用动量的优化器,如Adam,会丢失动量等状态信息,可能会造成损失函数的收敛出现震荡】
7. 模型训练? ? ? ? 我们通过调用optim实例的step函数来迭代模型参数。
? ? ? ? 输出:
下?面我们分别?比较学到的模型参数和真实的模型参数。我们从 net 获得需要的层,并访问其权重( weight )和偏差( bias )。学到的参数和真实的参数很接近
输出:
|
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 | -2024/11/27 15:48:34- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |