PyTorch如何搭建一个简单的网络


Posted in Python onAugust 24, 2020

1 任务

首先说下我们要搭建的网络要完成的学习任务: 让我们的神经网络学会逻辑异或运算,异或运算也就是俗称的“相同取0,不同取1” 。再把我们的需求说的简单一点,也就是我们需要搭建这样一个神经网络,让我们在输入(1,1)时输出0,输入(1,0)时输出1(相同取0,不同取1),以此类推。

2 实现思路

因为我们的需求需要有两个输入,一个输出,所以我们需要在输入层设置两个输入节点,输出层设置一个输出节点。因为问题比较简单,所以隐含层我们只需要设置10个节点就可以达到不错的效果了,隐含层的激活函数我们采用ReLU函数,输出层我们用Sigmoid函数,让输出保持在0到1的一个范围,如果输出大于0.5,即可让输出结果为1,小于0.5,让输出结果为0.

3 实现过程

我们使用的简单的快速搭建法。

3.1 引入必要库

import torch
import torch.nn as nn
import numpy as np

用pytorch当然要引入torch包,然后为了写代码方便将torch包里的nn用nn来代替,nn这个包就是neural network的缩写,专门用来搭神经网络的一个包。引入numpy是为了创建矩阵作为输入。

3.2 创建训练集

# 构建输入集
x = np.mat('0 0;'
      '0 1;'
      '1 0;'
      '1 1')
x = torch.tensor(x).float()
y = np.mat('1;'
      '0;'
      '0;'
      '1')
y = torch.tensor(y).float()

我个人比较喜欢用np.mat这种方式构建矩阵,感觉写法比较简单,当然你也可以用其他的方法。但是构建完矩阵一定要有这一步 torch.tensor(x).float() ,必须要把你所创建的输入转换成tensor变量。

什么是tensor呢?你可以简单地理解他就是pytorch中用的一种变量,你想用pytorch这个框架就必须先把你的变量转换成tensor变量。而我们这个神经网络会要求你的输入和输出必须是float浮点型的,指的是tensor变量中的浮点型,而你用np.mat创建的输入是int型的,转换成tensor也会自动地转换成tensor的int型,所以要在后面加个.float()转换成浮点型。

这样我们就构建完成了输入和输出(分别是x矩阵和y矩阵),x是四行二列的一个矩阵,他的每一行是一个输入,一次输入两个值,这里我们把所有的输入情况都列了出来。输出y是一个四行一列的矩阵,每一行都是一个输出,对应x矩阵每一行的输入。

3.3 搭建网络

myNet = nn.Sequential( 
  nn.Linear(2,10),
  nn.ReLU(),
  nn.Linear(10,1),
  nn.Sigmoid()
  )
print(myNet)

输出结果:

PyTorch如何搭建一个简单的网络

我们使用nn包中的Sequential搭建网络,这个函数就是那个可以让我们像搭积木一样搭神经网络的一个东西。

nn.Linear(2,10)的意思搭建输入层,里面的2代表输入节点个数,10代表输出节点个数。Linear也就是英文的线性,意思也就是这层不包括任何其它的激活函数,你输入了啥他就给你输出了啥。nn.ReLU()这个就代表把一个激活函数层,把你刚才的输入扔到了ReLU函数中去。 接着又来了一个Linear,最后再扔到Sigmoid函数中去。 2,10,1就分别代表了三个层的个数,简单明了。

3.4 设置优化器

optimzer = torch.optim.SGD(myNet.parameters(),lr=0.05)
loss_func = nn.MSELoss()

对这一步的理解就是,你需要有一个优化的方法来训练你的网络,所以这步设置了我们所要采用的优化方法。

torch.optim.SGD的意思就是采用SGD(随机梯度下降)方法训练,你只需要把你网络的参数和学习率传进去就可以了,分别是 myNet.paramets 和 lr 。 loss_func 这句设置了代价函数,因为我们的这个问题比较简单,所以采用了MSE,也就是均方误差代价函数。

3.5 训练网络

for epoch in range(5000):
  out = myNet(x)
  loss = loss_func(out,y)
  optimzer.zero_grad()
  loss.backward()
  optimzer.step()

我这里设置了一个5000次的循环(可能不需要这么多次),让这个训练的动作迭代5000次。每一次的输出直接用myNet(x),把输入扔进你的网络就得到了输出out(就是这么简单粗暴!),然后用代价函数和你的标准输出y求误差。 清除梯度的那一步是为了每一次重新迭代时清除上一次所求出的梯度,你就把这一步记住就行,初学不用理解太深。 loss.backward() 当然就是让误差反向传播,接着 optimzer.step() 也就是让我们刚刚设置的优化器开始工作。

3.6 测试

print(myNet(x).data)

运行结果:

PyTorch如何搭建一个简单的网络

可以看到这个结果已经非常接近我们期待的结果了,当然你也可以换个数据测试,结果也会是相似的。这里简单解释下为什么我们的代码末尾加上了一个.data,因为我们的tensor变量其实是包含两个部分的,一部分是tensor数据,另一部分是tensor的自动求导参数,我们加上.data意思是输出取tensor中的数据,如果不加的话会输出下面这样:

PyTorch如何搭建一个简单的网络

以上就是PyTorch如何搭建一个简单的网络的详细内容,更多关于PyTorch搭建网络的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
python分割文件的常用方法
Nov 01 Python
Python re模块介绍
Nov 30 Python
python将文本中的空格替换为换行的方法
Mar 19 Python
python中的变量如何开辟内存
Jun 26 Python
解决Mac下首次安装pycharm无project interpreter的问题
Oct 29 Python
python将txt文档每行内容循环插入数据库的方法
Dec 28 Python
python使用pymongo操作mongo的完整步骤
Apr 13 Python
Python玩转PDF的各种骚操作
May 06 Python
Django继承自带user表并重写的例子
Nov 18 Python
使用keras实现孪生网络中的权值共享教程
Jun 11 Python
Pycharm中使用git进行合作开发的教程详解
Nov 17 Python
PyCharm2020.3.2安装超详细教程
Feb 08 Python
Python pysnmp使用方法及代码实例
Aug 24 #Python
详解python tcp编程
Aug 24 #Python
Python rabbitMQ如何实现生产消费者模式
Aug 24 #Python
利用Python的folium包绘制城市道路图的实现示例
Aug 24 #Python
深入分析python 排序
Aug 24 #Python
超级实用的8个Python列表技巧
Aug 24 #Python
基于CentOS搭建Python Django环境过程解析
Aug 24 #Python
You might like
PHP+MySQL插入操作实例
2015/01/21 PHP
php flush无效,IIS7下php实时输出的方法
2016/08/25 PHP
利用PHP获取汉字首字母并且分组排序详解
2017/10/22 PHP
php报错502badgateway解决方法
2019/10/11 PHP
改变javascript函数内部this指针指向的三种方法
2010/04/23 Javascript
jquery插件制作教程 txtHover
2012/08/17 Javascript
js判断某个方法是否存在实例代码
2015/01/10 Javascript
JavaScript获取css行间样式,内连样式和外链样式的简单方法
2016/07/18 Javascript
浅谈JavaScript 函数参数传递到底是值传递还是引用传递
2016/08/23 Javascript
JavaScript字符串_动力节点Java学院整理
2017/06/27 Javascript
angularjs http与后台交互的实现示例
2018/12/21 Javascript
浅谈Vue服务端渲染框架Nuxt的那些事
2018/12/21 Javascript
vue自定义正在加载动画的例子
2019/11/14 Javascript
如何基于jQuery实现五角星评分
2020/09/02 jQuery
[55:35]DOTA2-DPC中国联赛 正赛 CDEC vs Dragon BO3 第二场 1月22日
2021/03/11 DOTA
基于Python的身份证号码自动生成程序
2014/08/15 Python
通过数据库对Django进行删除字段和删除模型的操作
2015/07/21 Python
关于python的bottle框架跨域请求报错问题的处理方法
2017/03/19 Python
Python中几种导入模块的方式总结
2017/04/27 Python
Python排序搜索基本算法之归并排序实例分析
2017/12/08 Python
flask使用session保存登录状态及拦截未登录请求代码
2018/01/19 Python
Python3实现的Mysql数据库操作封装类
2018/06/06 Python
ubuntu17.4下为python和python3装上pip的方法
2018/06/12 Python
Python线程障碍对象Barrier原理详解
2019/12/02 Python
python GUI库图形界面开发之PyQt5布局控件QHBoxLayout详细使用方法与实例
2020/03/06 Python
python+adb+monkey实现Rom稳定性测试详解
2020/04/23 Python
详解CSS3新增的背景属性
2019/12/25 HTML / CSS
美国婴儿和儿童家具网上商店:ABaby.com
2018/07/02 全球购物
师范毕业生自荐信
2013/10/17 职场文书
食品科学与工程专业毕业生求职信范文
2014/07/21 职场文书
大学生学习面向未来的赶考思想汇报
2014/09/12 职场文书
2014年后勤管理工作总结
2014/12/01 职场文书
英语教师求职信范文
2015/03/20 职场文书
分析JVM源码之Thread.interrupt系统级别线程打断
2021/06/29 Java/Android
用Java实现简单计算器功能
2021/07/21 Java/Android
PHP中国际化的字符串排序和比较对象详解
2021/08/23 PHP