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查找第k小元素代码分享
Dec 18 Python
python根据距离和时长计算配速示例
Feb 16 Python
Python中比较特别的除法运算和幂运算介绍
Apr 05 Python
在Python中使用全局日志时需要注意的问题
May 06 Python
python: 自动安装缺失库文件的方法
Oct 22 Python
python pandas cumsum求累计次数的用法
Jul 29 Python
Python如何根据时间序列数据作图
May 12 Python
解决keras backend 越跑越慢问题
Jun 18 Python
tensorflow下的图片标准化函数per_image_standardization用法
Jun 30 Python
详解python内置模块urllib
Sep 09 Python
Python实现自动装机功能案例分析
Oct 22 Python
详解pycharm连接远程linux服务器的虚拟环境的方法
Nov 13 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伪静态的实现详细介绍
2013/04/28 PHP
php中用date函数获取当前时间有误的解决办法
2013/08/02 PHP
基于PHP生成简单的验证码
2016/06/01 PHP
TP3.2框架分页相关实现方法分析
2020/06/03 PHP
ExtJs3.0中Store添加 baseParams 的Bug
2010/03/10 Javascript
jQuery之浮动窗口实现代码(两种方法)
2010/09/08 Javascript
javascript一些实用技巧小结
2011/03/18 Javascript
js跑步算法的实现代码
2013/12/04 Javascript
JQuery中dataGrid设置行的高度示例代码
2014/01/03 Javascript
javascript中键盘事件用法实例分析
2015/01/30 Javascript
Javascript中获取对象的原型对象的方法小结
2015/02/25 Javascript
JS中把函数作为另一函数的参数传递方法(总结)
2017/06/28 Javascript
jQuery使用zTree插件实现可拖拽的树示例
2017/09/23 jQuery
浅谈react-native热更新react-native-pushy集成遇到的问题
2017/09/30 Javascript
vue中实现滚动加载更多的示例
2017/11/08 Javascript
jquery 动态遍历select 赋值的实例
2018/09/12 jQuery
js删除数组中某几项的方法总结
2019/01/16 Javascript
如何在VUE中使用vue-awesome-swiper
2021/01/04 Vue.js
零基础写python爬虫之神器正则表达式
2014/11/06 Python
Python描述器descriptor详解
2015/02/03 Python
举例讲解Python中的list列表数据结构用法
2016/03/12 Python
python批量替换多文件字符串问题详解
2018/04/22 Python
Python实现调用另一个路径下py文件中的函数方法总结
2018/06/07 Python
Pandas DataFrame数据的更改、插入新增的列和行的方法
2019/06/25 Python
python全栈知识点总结
2019/07/01 Python
pycharm 对代码做静态检查操作
2020/06/09 Python
解决Windows下python和pip命令无法使用的问题
2020/08/31 Python
科沃斯机器人官网商城:Ecovacs
2016/08/29 全球购物
澳大利亚最大的女装零售商:Millers
2017/09/10 全球购物
联想瑞士官方网站:Lenovo Switzerland
2017/11/19 全球购物
电气工程和自动化自荐信范文
2013/12/25 职场文书
亮化工程实施方案
2014/03/17 职场文书
2014年新生军训方案
2014/05/01 职场文书
湖南省召开党的群众路线教育实践活动总结大会报告
2014/10/21 职场文书
新员工入职欢迎词
2015/01/23 职场文书
Python与C++中梯度方向直方图的实现
2022/03/17 Python