738 B
738 B
Usage of optimizers
An optimizer is one of the two arguments required for compiling a Keras model:
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.
# pass optimizer by name: default parameters will be used
model.compile(loss='mean_squared_error', optimizer='sgd')
{{autogenerated}}