python生成线性回归数据

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()

image.png

未经允许不得转载!python生成线性回归数据