线性回归的简洁实现

最新推荐文章于 2024-07-09 17:55:25 发布

Nlper_Zhao

最新推荐文章于 2024-07-09 17:55:25 发布

阅读量203

点赞数

分类专栏：深度学习和神经网络文章标签：深度学习神经网络 pytorch

本文链接：https://blog.csdn.net/weixin_40605573/article/details/108999480

版权

import torch
import numpy as np
import torch.utils.data as Data
import torch.nn as nn
from torch.nn import init
import torch.optim as optim

# 1、生成数据集
num_inputs = 2
num_examples = 1000
true_w = [2,-3.4]
true_b = 4.2
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