adagrad-gluon.md 1.5 KB
Newer Older
1 2 3 4 5
# Adagrad --- 使用Gluon


`Gluon`里,使用Adagrad很容易。我们无需重新实现它。

A
Aston Zhang 已提交
6 7 8
```{.python .input}
%config InlineBackend.figure_format = 'retina'
%matplotlib inline
9 10 11
import mxnet as mx
from mxnet import autograd
from mxnet import gluon
A
Aston Zhang 已提交
12
from mxnet import nd
A
Aston Zhang 已提交
13 14 15 16 17
import numpy as np
import sys
sys.path.append('..')
import utils
```
18

A
Aston Zhang 已提交
19
```{.python .input  n=1}
20 21 22 23 24 25 26 27 28
# 生成数据集。
num_inputs = 2
num_examples = 1000
true_w = [2, -3.4]
true_b = 4.2
X = nd.random_normal(scale=1, shape=(num_examples, num_inputs))
y = true_w[0] * X[:, 0] + true_w[1] * X[:, 1] + true_b
y += .01 * nd.random_normal(scale=1, shape=y.shape)

A
Aston Zhang 已提交
29
# 线性回归模型。
30 31 32 33 34 35 36 37 38
net = gluon.nn.Sequential()
net.add(gluon.nn.Dense(1))
```

我们需要在`gluon.Trainer`中指定优化算法名称`adagrad`并设置参数。例如设置初始学习率`learning_rate`

使用Adagrad,最终学到的参数值与真实值较接近。

```{.python .input  n=3}
A
Aston Zhang 已提交
39 40 41 42 43
net.collect_params().initialize(mx.init.Normal(sigma=1), force_reinit=True)
trainer = gluon.Trainer(net.collect_params(), 'adagrad',
                        {'learning_rate': 0.9})
utils.optimize(batch_size=10, trainer=trainer, num_epochs=3, decay_epoch=None,
               log_interval=10, X=X, y=y, net=net)
44 45
```

A
Aston Zhang 已提交
46
## 小结
47

A
Aston Zhang 已提交
48
* 使用`Gluon``Trainer`可以方便地使用Adagrad。
49 50 51 52 53

## 练习

* 尝试使用其他的初始学习率,结果有什么变化?

A
Aston Zhang 已提交
54 55 56 57 58
## 讨论

欢迎扫码直达[本节内容讨论区](https://discuss.gluon.ai/t/topic/2274)

![](../img/qr_adagrad-gluon.svg)