candle.keras_utils.build_optimizer

candle.keras_utils.build_optimizer(optimizer, lr, kerasDefaults)

Set the optimizer to the appropriate Keras optimizer function based on the input string and learning rate. Other required values are set to the Keras default values.

Parameters:

• optimizer (string) – String to choose the optimizer Options recognized: ‘sgd’, ‘rmsprop’, ‘adagrad’, adadelta’, ‘adam’ See the Keras documentation for a full description of the options

• lr (float) – Learning rate

• kerasDefaults (Dict) – Dictionary of default parameter values to ensure consistency between frameworks

Returns:

The appropriate Keras optimizer function