python导出数据成mat_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()

本文地址：https://ai.52learn.online/843