25 lines
738 B
Markdown
25 lines
738 B
Markdown
|
|
||
|
|
## Usage of optimizers
|
||
|
|
|
||
|
|
An optimizer is one of the two arguments required for compiling a Keras model:
|
||
|
|
|
||
|
|
```python
|
||
|
|
model = Sequential()
|
||
|
|
model.add(Dense(64, init='uniform', input_dim=10))
|
||
|
|
model.add(Activation('tanh'))
|
||
|
|
model.add(Activation('softmax'))
|
||
|
|
|
||
|
|
sgd = SGD(lr=0.1, decay=1e-6, momentum=0.9, nesterov=True)
|
||
|
|
model.compile(loss='mean_squared_error', optimizer=sgd)
|
||
|
|
```
|
||
|
|
|
||
|
|
You can either instantiate an optimizer before passing it to `model.compile()` , as in the above example, or you can call it by its name. In the latter case, the default parameters for the optimizer will be used.
|
||
|
|
|
||
|
|
```python
|
||
|
|
# pass optimizer by name: default parameters will be used
|
||
|
|
model.compile(loss='mean_squared_error', optimizer='sgd')
|
||
|
|
```
|
||
|
|
|
||
|
|
---
|
||
|
|
|
||
|
|
{{autogenerated}}
|