org.deeplearning4j.nn.modelimport.keras.utils

Class KerasOptimizerUtils

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.utils.KerasOptimizerUtils

public class KerasOptimizerUtils
extends Object

Utility functionality for keras optimizers

Author:

Max Pumperla

Constructor Summary

Constructors

Constructor and Description
KerasOptimizerUtils()

Method Summary

Modifier and Type	Method and Description
static IUpdater	mapOptimizer(Map<String,Object> optimizerConfig) Map Keras optimizer to DL4J IUpdater.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KerasOptimizerUtils

public KerasOptimizerUtils()

Method Detail

mapOptimizer

public static IUpdater mapOptimizer(Map<String,Object> optimizerConfig)
                             throws UnsupportedKerasConfigurationException,
                                    InvalidKerasConfigurationException

Map Keras optimizer to DL4J IUpdater.

Parameters:

optimizerConfig - Optimizer configuration map

Returns:

DL4J IUpdater instance

Throws:

UnsupportedKerasConfigurationException

InvalidKerasConfigurationException