org.deeplearning4j.nn.modelimport.keras.utils

Class KerasOptimizerUtils

java.lang.Object

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

public class KerasOptimizerUtils
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected static String	BETA_1
protected static String	BETA_2
protected static String	DECAY
protected static String	EPSILON
protected static String	LR
protected static String	LR2
protected static String	MOMENTUM
protected static String	RHO
protected static String	SCHEDULE_DECAY

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

Field Detail

LR

protected static final String LR

See Also:

Constant Field Values

LR2

protected static final String LR2

See Also:

Constant Field Values

EPSILON

protected static final String EPSILON

See Also:

Constant Field Values

MOMENTUM

protected static final String MOMENTUM

See Also:

Constant Field Values

BETA_1

protected static final String BETA_1

See Also:

Constant Field Values

BETA_2

protected static final String BETA_2

See Also:

Constant Field Values

DECAY

protected static final String DECAY

See Also:

Constant Field Values

RHO

protected static final String RHO

See Also:

Constant Field Values

SCHEDULE_DECAY

protected static final String SCHEDULE_DECAY

See Also:

Constant Field Values

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