深度学习入门,自动生成数据训练线性回归模型

本文介绍了如何使用PyTorch生成y=ax+by+c格式的数据,并训练线性回归模型。通过随机生成数据,利用DataLoader读取数据集,定义并初始化模型,选择MSELoss作为损失函数,采用SGD优化算法进行训练,最终比较真实参数与模型学习到的参数。
摘要由CSDN通过智能技术生成

针对y=ax+.by+c 格式 首先自动生成一千个数据,然后使用模型训练。代码复制即可运行。
import torch
from IPython import display
from matplotlib import pyplot as plt
import numpy as np
import random
import torch.utils.data as Data
from torch import nn
from torch.nn import init
import torch.optim as optim
#shengchengshujuji
num_inputs = 2
num_examples = 1000
true_b = 3.2
true_w = [1.2,-2.4]
features = torch.tensor(np.random.normal(0,1,(num_examples,num_inputs)),dtype=torch.float)
labels = true_w[0] * features[:,0] + true_w[1] * features[:,1] + true_b
labels += torch.tensor(np.random.normal(0,0.01,size=labels.size()),dtype=torch.float)
#读取数据集
batch_size = 10
dataset = Data.TensorDataset(features,labels)
data_iter = Data.DataLoader(dataset,batch_size,shuffle=True)
#定义模型
net = nn.Sequential(
nn.L

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值