pytorch-神经网络拟合曲线实例


Posted in Python onJanuary 15, 2020

代码已经调通,跑出来的效果如下:

pytorch-神经网络拟合曲线实例

# coding=gbk
import torch
import matplotlib.pyplot as plt
from torch.autograd import Variable
import torch.nn.functional as F
 
'''
 Pytorch是一个拥有强力GPU加速的张量和动态构建网络的库,其主要构建是张量,所以可以把PyTorch当做Numpy
 来用,Pytorch的很多操作好比Numpy都是类似的,但是其能够在GPU上运行,所以有着比Numpy快很多倍的速度。
 训练完了,发现隐层越大,拟合的速度越是快,拟合的效果越是好
'''
 
def train():
 print('------  构建数据集  ------')
 # torch.linspace是为了生成连续间断的数据,第一个参数表示起点,第二个参数表示终点,第三个参数表示将这个区间分成平均几份,即生成几个数据
 x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
 #torch.rand返回的是[0,1]之间的均匀分布 这里是使用一个计算式子来构造出一个关联结果,当然后期要学的也就是这个式子
 y = x.pow(2) + 0.2 * torch.rand(x.size())
 # Variable是将tensor封装了下,用于自动求导使用
 x, y = Variable(x), Variable(y)
 #绘图展示
 plt.scatter(x.data.numpy(), y.data.numpy())
 #plt.show()
 
 print('------  搭建网络  ------')
 #使用固定的方式继承并重写 init和forword两个类
 class Net(torch.nn.Module):
  def __init__(self,n_feature,n_hidden,n_output):
   #初始网络的内部结构
   super(Net,self).__init__()
   self.hidden=torch.nn.Linear(n_feature,n_hidden)
   self.predict=torch.nn.Linear(n_hidden,n_output)
  def forward(self, x):
   #一次正向行走过程
   x=F.relu(self.hidden(x))
   x=self.predict(x)
   return x
 net=Net(n_feature=1,n_hidden=1000,n_output=1)
 print('网络结构为:',net)
 
 print('------  启动训练  ------')
 loss_func=F.mse_loss
 optimizer=torch.optim.SGD(net.parameters(),lr=0.001)
 
 #使用数据 进行正向训练,并对Variable变量进行反向梯度传播 启动100次训练
 for t in range(10000):
  #使用全量数据 进行正向行走
  prediction=net(x)
  loss=loss_func(prediction,y)
  optimizer.zero_grad() #清除上一梯度
  loss.backward() #反向传播计算梯度
  optimizer.step() #应用梯度
 
  #间隔一段,对训练过程进行可视化展示
  if t%5==0:
   plt.cla()
   plt.scatter(x.data.numpy(),y.data.numpy()) #绘制真是曲线
   plt.plot(x.data.numpy(),prediction.data.numpy(),'r-',lw=5)
   plt.text(0.5,0,'Loss='+str(loss.data[0]),fontdict={'size':20,'color':'red'})
   plt.pause(0.1)
 plt.ioff()
 plt.show()
 print('------  预测和可视化  ------')
 
if __name__=='__main__':
 train()

以上这篇pytorch-神经网络拟合曲线实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 中文字符串的处理实现代码
Oct 25 Python
python字符串替换示例
Apr 24 Python
在Python的Tornado框架中实现简单的在线代理的教程
May 02 Python
Python定时器实例代码
Nov 01 Python
Python实现按特定格式对文件进行读写的方法示例
Nov 30 Python
python正则-re的用法详解
Jul 28 Python
pytorch 使用加载训练好的模型做inference
Feb 20 Python
Python Tornado实现WEB服务器Socket服务器共存并实现交互的方法
May 26 Python
Python使用lambda抛出异常实现方法解析
Aug 20 Python
Python中Yield的基本用法
Oct 18 Python
使用python+pygame开发消消乐游戏附完整源码
Jun 10 Python
使用python将HTML转换为PDF pdfkit包(wkhtmltopdf) 的使用方法
Apr 21 Python
Pytorch中的VGG实现修改最后一层FC
Jan 15 #Python
详解Python3 中的字符串格式化语法
Jan 15 #Python
用pytorch的nn.Module构造简单全链接层实例
Jan 14 #Python
pytorch三层全连接层实现手写字母识别方式
Jan 14 #Python
Python实现bilibili时间长度查询的示例代码
Jan 14 #Python
基于python监控程序是否关闭
Jan 14 #Python
关于pytorch中全连接神经网络搭建两种模式详解
Jan 14 #Python
You might like
php $_SERVER["REQUEST_URI"]获取值的通用解决方法
2010/06/21 PHP
PHP开发负载均衡指南
2010/07/17 PHP
使用php判断浏览器的类型和语言的函数代码
2013/02/28 PHP
php提交post数组参数实例分析
2015/12/17 PHP
PHP处理CSV表格文件的常用操作方法总结
2016/07/01 PHP
PHP与服务器文件系统的简单交互
2016/10/21 PHP
PHP实现的数独求解问题示例
2017/04/18 PHP
PHP生成随机码的思路与方法实例探索
2019/04/11 PHP
jquery offset函数应用实例
2012/11/14 Javascript
js 删除数组的几种方法小结
2014/02/21 Javascript
表单提交前触发函数返回true表单才会提交
2014/03/11 Javascript
基于insertBefore制作简单的循环插空效果
2015/09/21 Javascript
JavaScript实现ASC转汉字及汉字转ASC的方法
2016/01/23 Javascript
jQuery实现的省市县三级联动菜单效果完整实例
2016/08/01 Javascript
微信小程序-消息提示框实例
2016/11/24 Javascript
vue过渡和animate.css结合使用详解
2017/06/14 Javascript
JavaScript的Object.defineProperty详解
2018/07/09 Javascript
点击按钮弹出模态框的一系列操作代码实例
2019/03/29 Javascript
微信小程序如何利用getCurrentPages进行页面传值
2019/07/01 Javascript
javascript json对象小技巧之键名作为变量用法分析
2019/11/11 Javascript
Vue 如何使用props、emit实现自定义双向绑定的实现
2020/06/05 Javascript
js实现贪吃蛇小游戏(加墙)
2020/07/31 Javascript
vue 在单页面应用里使用二级套嵌路由
2020/12/19 Vue.js
浅谈MySQL中的触发器
2015/05/05 Python
PyCharm搭建Spark开发环境的实现步骤
2019/09/05 Python
python tkinter之顶层菜单、弹出菜单实例
2020/03/04 Python
python GUI库图形界面开发之PyQt5 MDI(多文档窗口)QMidArea详细使用方法与实例
2020/03/05 Python
解决Keras TensorFlow 混编中 trainable=False设置无效问题
2020/06/28 Python
Traffic People官网:女式花裙、上衣和连身裤
2020/10/12 全球购物
自荐信格式
2013/12/01 职场文书
给面试官的感谢信
2014/02/01 职场文书
高中学生干部学习的自我评价
2014/02/21 职场文书
公司总经理工作职责管理办法
2014/02/28 职场文书
见习报告的格式
2014/10/31 职场文书
2015年农村党员干部主题教育活动总结
2015/03/25 职场文书
HR必备:超全面的薪酬待遇管理方案!
2019/07/12 职场文书