import torch from matplotlib import pyplot as plt import numpy as np import random #生成数据集 num_inputs = 2 num_examples = 1000 true_w = [2, -3.4] true_b = 4.2 features = torch.randn(num_examples, num_inputs, dtype=torch.float32) 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.float32) print(features[0], labels[0]) #显示数据集 plt.scatter(features[:, 1].numpy(), labels.numpy(), 1); plt.show()
未经允许不得转载!python生成线性回归数据