org.deeplearning4j.nn.modelimport.keras.utils

Class KerasConstraintUtils

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.utils.KerasConstraintUtils

public class KerasConstraintUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
KerasConstraintUtils()

Method Summary

Modifier and Type	Method and Description
static LayerConstraint	getConstraintsFromConfig(Map<String,Object> layerConfig, String constraintField, KerasLayerConfiguration conf, int kerasMajorVersion) Get constraint initialization from Keras layer configuration.
static LayerConstraint	mapConstraint(String kerasConstraint, KerasLayerConfiguration conf, Map<String,Object> constraintConfig) Map Keras to DL4J constraint.

Methods inherited from class java.lang.Object

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

Constructor Detail

KerasConstraintUtils

public KerasConstraintUtils()

Method Detail

mapConstraint

public static LayerConstraint mapConstraint(String kerasConstraint,
                                            KerasLayerConfiguration conf,
                                            Map<String,Object> constraintConfig)
                                     throws UnsupportedKerasConfigurationException

Map Keras to DL4J constraint.

Parameters:

kerasConstraint - String containing Keras constraint name

conf - Keras layer configuration

Returns:

DL4J LayerConstraint

Throws:

UnsupportedKerasConfigurationException

See Also:

LayerConstraint

getConstraintsFromConfig

public static LayerConstraint getConstraintsFromConfig(Map<String,Object> layerConfig,
                                                       String constraintField,
                                                       KerasLayerConfiguration conf,
                                                       int kerasMajorVersion)
                                                throws InvalidKerasConfigurationException,
                                                       UnsupportedKerasConfigurationException

Get constraint initialization from Keras layer configuration.

Parameters:

layerConfig - dictionary containing Keras layer configuration

constraintField - string in configuration representing parameter to constrain

conf - Keras layer configuration

kerasMajorVersion - Major keras version as integer (1 or 2)

Returns:

a valid LayerConstraint

Throws:

InvalidKerasConfigurationException - Invalid configuration

UnsupportedKerasConfigurationException - Unsupported configuration