当前位置:网站首页>Station B Liu Erden linear regression pytoch
Station B Liu Erden linear regression pytoch
2022-07-06 05:41:00 【Ning Ranye】
Series articles :
List of articles
Linear regression Pytorch
import matplotlib.pyplot as plt
import torch
x_data = torch.Tensor([[1.0, 2.0, 3.0]])
y_data = torch.Tensor([[2.0, 4.0, 6.0]])
class LinearModel(torch.nn.Module):
# Member method __init__ And forward Must be realized
def __init__(self):
super(LinearModel, self).__init__()
# 1,1, For weight weight And offset b Of size
self.linear = torch.nn.Linear(3, 1)
def forward(self, x):
y_pred = self.linear(x)
return y_pred
model = LinearModel()
# Loss function calculation one batch All in loss The average of
criterion = torch.nn.MSELoss(size_average=False)
optimizer = torch.optim.SGD(model.parameters(), lr=0.0005)
loss_list = []
epoch_list = []
for epoch in range(200):
y_pred = model(x_data)
loss = criterion(y_pred, y_data)
print(epoch, loss.item())
loss_list.append(loss.item())
epoch_list.append(epoch)
# The gradient will be backward Hourly accumulation , So in backward Must be reset before
optimizer.zero_grad()
loss.backward()
# update
optimizer.step()
plt.plot(epoch_list, loss_list)
plt.xlabel("epoch")
plt.ylabel("loss")
plt.show()
边栏推荐
- Modbus protocol communication exception
- B站刘二大人-反向传播
- Jvxetable用slot植入j-popup
- The ECU of 21 Audi q5l 45tfsi brushes is upgraded to master special adjustment, and the horsepower is safely and stably increased to 305 horsepower
- Sword finger offer II 039 Maximum rectangular area of histogram
- A master in the field of software architecture -- Reading Notes of the beauty of Architecture
- Codeless June event 2022 codeless Explorer conference will be held soon; AI enhanced codeless tool launched
- Zoom and pan image in Photoshop 2022
- Summary of deep learning tuning tricks
- js Array 列表 实战使用总结
猜你喜欢
The digital economy has broken through the waves. Is Ltd a Web3.0 website with independent rights and interests?
How can large websites choose better virtual machine service providers?
应用安全系列之三十七:日志注入
ArcGIS应用基础4 专题图的制作
04. Project blog log
03. 开发博客项目之登录
Migrate Infones to stm32
01. Project introduction of blog development project
js Array 列表 实战使用总结
Sequoiadb Lake warehouse integrated distributed database, June 2022 issue
随机推荐
Remember an error in MySQL: the user specified as a definer ('mysql.infoschema '@' localhost ') does not exist
什么是独立IP,独立IP主机怎么样?
Rustdesk builds its own remote desktop relay server
Safe mode on Windows
实践分享:如何安全快速地从 Centos迁移到openEuler
Notes, continuation, escape and other symbols
【经验】UltralSO制作启动盘时报错:磁盘/映像容量太小
【SQL server速成之路】——身份验证及建立和管理用户账户
【torch】|torch.nn.utils.clip_grad_norm_
Application Security Series 37: log injection
UCF (summer team competition II)
ArcGIS应用基础4 专题图的制作
Problems encountered in installing mysql8 on MAC
UCF (2022 summer team competition I)
嵌入式面试题(一:进程与线程)
Pytorch代码注意的细节,容易敲错的地方
The ECU of 21 Audi q5l 45tfsi brushes is upgraded to master special adjustment, and the horsepower is safely and stably increased to 305 horsepower
2022半年总结
HAC cluster modifying administrator user password
A master in the field of software architecture -- Reading Notes of the beauty of Architecture