gluon学习系列1 线性回归 — 从0开始1

线性回归 — 从0开始

错误修正前
from mxnet import ndarray
from mxnet import autograd
import matplotlib.pyplot as plt
true_w = [-2, 5]
true_b = 1.5
num_batch = 1000
X = ndarray.random_normal(shape=[num_batch, 2])
y = ndarray.ones(shape=[num_batch, 1])
y = X[:, 0]*true_w[0]+X[:, 1]*true_w[1]+true_b

其中groundtrue的写法一般写成true_x。如true_w
数值型的变量一般用num_x。训练集写成num_examples
ndarray.random_normal生成正态分布
ndarray.ondes生成全1矩阵,上面可以不用给y初始化。

修订后代码
from mxnet import ndarray
from mxnet import autograd
import matplotlib.pyplot as plt
true_w = [-2, 5]
true_b = 1.5
num_inputs = 2
num_examples = 1000
X = ndarray.random_normal(shape=[num_examples, num_inputs])
y = X[:, 0]*true_w[0]+X[:, 1]*true_w[1]+true_b

print(X[0:5],y[0:5])

[[ 1.16307855  0.48380461]
 [ 0.29956347  0.15302546]
 [-1.16881478  1.55807102]
 [-0.54594457 -2.35562968]
 [ 0.54144025  2.67850637]]
<NDArray 5x2 @cpu(0)> 
[  1.59286594   1.66600037  11.627985    -9.18625927  13.80965233]
<NDArray 5 @cpu(0)>
plt.scatter(X[:, 1].asnumpy(), y.asnumpy())
plt.show()

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值