基于mxnet框架的线性回归简洁实现

python代码

#生成数据集
from mxnet import autograd,nd
num_inputs=2
num_examples=1000
true_w=[2,-3,4]
true_b=4.2
features=nd.random.normal(scale=1,shape=(num_examples,num_inputs))
labels=true_w[0]*features[:,0]+true_w[1]*features[:,1]+true_b
labels+=nd.random.normal(scale=0.01,shape=labels.shape)


#读取数据集
from mxnet.gluon import data as gdata
batch_size=10
dataset=gdata.ArrayDataset(features,labels)
data_iter=gdata.DataLoader(dataset,batch_size,shuffle=True)
for x,y in data_iter:
    print(x,y)
    break


#定义模型
from mxnet.gluon import nn
net=nn.Sequential()
net.add(nn.Dense(1))


#初始化模型参数
from mxnet import init
net.initialize(init.Normal(sigma=0.01))


#定义损失函数
from mxnet.gluon import loss as gloss
loss=gloss.L2Loss()


#定义优化算法
from mxnet import gluon
trainer=gluon.Trainer(net.collect_params(),'sgd',{'learning_rate':0.03})


#训练模型
num_epochs=3
for epoch in range(1,num_epochs+1):
    for x,y in data_iter:
        with autograd.record():
            l=loss(net(x),y)
        l.backward()
        trainer.step(batch_size)
    l=loss(net(features),labels)
    print('epoch %d,loss:%f'%(epoch,l.mean().asnumpy()))
#输出权重和偏差
dense=net[0]
print(true_w,dense.weight.data())
print(true_b,dense.bias.data())

运行结果