当前位置:网站首页>B站刘二大人-线性回归 Pytorch
B站刘二大人-线性回归 Pytorch
2022-07-06 05:33:00 【宁然也】
系列文章:
文章目录
线性回归 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):
# 成员方法__init__ 及 forward 必须实现
def __init__(self):
super(LinearModel, self).__init__()
# 1,1,代表权重weight和偏置b的size
self.linear = torch.nn.Linear(3, 1)
def forward(self, x):
y_pred = self.linear(x)
return y_pred
model = LinearModel()
# 损失函数计算一个batch中所有loss的均值
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)
# 梯度会在backward时累计,所以在backward之前必须重置
optimizer.zero_grad()
loss.backward()
# update
optimizer.step()
plt.plot(epoch_list, loss_list)
plt.xlabel("epoch")
plt.ylabel("loss")
plt.show()
边栏推荐
- Using stopwatch to count code time
- 应用安全系列之三十七:日志注入
- Huawei equipment is configured with OSPF and BFD linkage
- Summary of redis basic knowledge points
- [leetcode16] the sum of the nearest three numbers (double pointer)
- 剑指 Offer II 039. 直方图最大矩形面积
- [untitled]
- Zoom and pan image in Photoshop 2022
- Promise summary
- Closure, decorator
猜你喜欢
nacos-高可用seata之TC搭建(02)
Easy to understand I2C protocol
【torch】|torch.nn.utils.clip_grad_norm_
Problems encountered in installing mysql8 on MAC
Graduation design game mall
Steady, 35K, byte business data analysis post
PDK工艺库安装-CSMC
[force buckle]43 String multiplication
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
[cloud native] 3.1 kubernetes platform installation kubespher
随机推荐
How to get list length
Can the feelings of Xi'an version of "Coca Cola" and Bingfeng beverage rush for IPO continue?
Deep learning -yolov5 introduction to actual combat click data set training
指针经典笔试题
Text classification still stays at Bert? The dual contrast learning framework is too strong
巨杉数据库再次亮相金交会,共建数字经济新时代
Web Security (V) what is a session? Why do I need a session?
Talking about the type and function of lens filter
03. Login of development blog project
Remember an error in MySQL: the user specified as a definer ('mysql.infoschema '@' localhost ') does not exist
Sword finger offer II 039 Maximum rectangular area of histogram
Mysql高级篇学习总结9:创建索引、删除索引、降序索引、隐藏索引
HAC集群修改管理员用户密码
Easy to understand I2C protocol
Cuda11.1 online installation
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
Excel转换为Lua的配置文件
Vulhub vulnerability recurrence 67_ Supervisor
Yyds dry inventory SSH Remote Connection introduction
Vulhub vulnerability recurrence 68_ ThinkPHP