机器学习笔记180328-监督学习-线性回归————使用Gluon

机器学习笔记180328-监督学习-线性回归——使用Gluon

1.创建数据集

from mxnet import ndarray as nd
from mxnet import autograd
from mxnet import gluon

num_inputs = 2
num_examples = 1000

true_w = [2, -3.4]
true_b = 4.2

X = nd.random_normal(shape=(num_examples, num_inputs))
y = true_w[0] * X[:, 0] + true_w[1] * X[:, 1] + true_b
y += .01 * nd.random_normal(shape=y.shape) #加噪音#

2.数据读取

batch_size = 10 
dataset = gluon.data.ArrayDataset(X, y)
data_iter = gluon.data.DataLoader(dataset, batch_size, shuffle=True)

for data, label in data_iter:
    print(data, label)
    break

    关于epoch、iteration、batchsize参看:    https://blog.csdn.net/sinat_30071459/article/details/50721