import torch from IPython import displayfrom matplotlib import pyplot as pltimport numpy as npimport random# 生成数据集num_inputs =2 # 两个变量num_examples = 1000true_w = [2, -3.4]true_b = 4.2# numpy.random.normal(loc=0.0, scale=1.0, size=None)# loc:float此概率分布的均值(对应着整个分布的中心centre)#scale:float 此概率分布的标准差(对应于分布的宽度,scale越大越矮胖,scale越小,越瘦高)#size:int or tuple of ints输出的shape,默认为None,只输出一个值features = torch.from_numpy(np.random.normal(0,1,(num_examples, num_inputs))) #2行1000列labels = true_w[0]*features[:,0]+true_w[1]*features[:,1]+true_b # 预测值= w1*x1+w2*x2+blabels += torch.from_numpy(np.random.normal(0, 0.01, size=labels.size()))print( labels.size()) def use_svg_display():# ⽤用⽮矢量量图显示 display.set_matplotlib_formats('svg')def set_figsize(figsize
线性回归数据集_动手学pytorchday2线性回归生成数据集
最新推荐文章于 2024-06-10 11:08:19 发布