Function getOptimizer

tfjs-layers/src/optimizers.ts:25–45

(identifier: string)

// Porting note: This diverges from the PyKeras implementation and may need to
// change based on (de)serialization requirements.
export function getOptimizer(identifier: string): Optimizer {
  const optimizerMap: {[optimizerName: string]: () => Optimizer} = {
    'Adagrad': () => train.adagrad(0.01),
    'Adadelta': () => train.adadelta(1, 0.95, epsilon()),
    'Adam': () => train.adam(0.001, 0.9, 0.999, epsilon()),
    'Adamax': () => train.adamax(0.002, 0.9, 0.999, epsilon(), 0),
    'RMSProp': () => train.rmsprop(0.001, 0.9, 0, epsilon()),
    'SGD': () => train.sgd(0.01)
  };
  optimizerMap['adagrad'] = optimizerMap['Adagrad'];
  optimizerMap['adadelta'] = optimizerMap['Adadelta'];
  optimizerMap['adam'] = optimizerMap['Adam'];
  optimizerMap['adamax'] = optimizerMap['Adamax'];
  optimizerMap['rmsprop'] = optimizerMap['RMSProp'];
  optimizerMap['sgd'] = optimizerMap['SGD'];

  if (identifier in optimizerMap) {
    return optimizerMap[identifier]();
  }
  throw new ValueError(`Unknown Optimizer ${identifier}`);
}