pytorch入门学习（三）-----线性回归模型

最新推荐文章于 2024-08-07 14:10:52 发布

weixin_44776845

最新推荐文章于 2024-08-07 14:10:52 发布

阅读量164

点赞数

分类专栏： pytorch 文章标签： python pytorch pycharm

本文链接：https://blog.csdn.net/weixin_44776845/article/details/104584473

版权

本文介绍PyTorch中线性回归模型的学习，包括如何导入matplotlib，调整学习率，创建不同规模的训练数据，以及设置不同的斜率和截距。适合初学者入门。

摘要由CSDN通过智能技术生成

在运行之前需要matplotlib包，快速方便的方法可以看这个pycharm导入matplotlib

可以尝试设置不同的学习率，创建不同大小的训练数据集，以及不同的斜率和截距

import torch
import matplotlib.pyplot as plt
torch.manual_seed(10)

# 设定学习率为0.01
lr = 0.01
best_loss = float("inf")


# 创建训练数据集，
# 即制造出接近 y = 4x + 6 的数据集，加上torch.randn()制造噪声
x = torch.rand(100, 1) * 10
y = 4*x + (6 + torch.randn(100, 1))

# 构