pytorch-神经网络拟合曲线实例
作者:马飞飞 时间:2022-03-17 18:17:30
代码已经调通,跑出来的效果如下:
# 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()
来源:https://blog.csdn.net/maqunfi/article/details/84504622
标签:pytorch,神经网络,拟合曲线
0
投稿
猜你喜欢
用python实现学生管理系统
2021-05-22 12:13:31
python 查找文件名包含指定字符串的方法
2021-05-03 08:54:21
CSS的渲染效率:书写高效的CSS
2008-11-13 16:55:00
Python执行ping操作的简单方法
2022-09-22 12:09:22
uni-app网络请求、数据缓存实例详解
2023-08-09 03:49:12
Python中如何给字典设置默认值
2023-09-21 00:15:32
window.open被浏览器拦截后的自定义提示
2007-11-23 12:31:00
关于python写入文件自动换行的问题
2021-09-01 00:03:13
js选取多个或单个元素的实现代码(用class)
2023-08-24 16:51:29
python中字典增加和删除使用方法
2023-12-15 15:31:30
python实现不同电脑之间视频传输功能
2021-01-07 00:03:40
Python识别html主要文本框过程解析
2023-11-09 01:31:52
Dreamweaver制作网页11种技巧详解
2007-11-05 14:00:00
Python读取xlsx文件报错:xlrd.biffh.XLRDError: Excel xlsx file;not supported问题解决
2021-02-02 16:08:59
python实现宿舍管理系统
2023-12-28 14:14:27
Django中间件拦截未登录url实例详解
2022-05-15 05:30:21
在VS Code上搭建Python开发环境的方法
2021-02-19 14:20:38
Python 如何修改程序默认时区
2022-10-07 15:32:16
python爬取招聘要求等信息实例
2021-01-27 21:22:36
巧妙的自关联运用
2012-10-07 10:55:58