org.deeplearning4j.nn.modelimport.keras.utils

Class KerasInitilizationUtils

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.utils.KerasInitilizationUtils

public class KerasInitilizationUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
KerasInitilizationUtils()

Method Summary

Modifier and Type	Method and Description
static IWeightInit	getWeightInitFromConfig(Map<String,Object> layerConfig, String initField, boolean enforceTrainingConfig, KerasLayerConfiguration conf, int kerasMajorVersion) Get weight initialization from Keras layer configuration.
static IWeightInit	mapWeightInitialization(String kerasInit, KerasLayerConfiguration conf, Map<String,Object> initConfig, int kerasMajorVersion) Map Keras to DL4J weight initialization functions.

Methods inherited from class java.lang.Object

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

Constructor Detail

KerasInitilizationUtils

public KerasInitilizationUtils()

Method Detail

mapWeightInitialization

public static IWeightInit mapWeightInitialization(String kerasInit,
                                                  KerasLayerConfiguration conf,
                                                  Map<String,Object> initConfig,
                                                  int kerasMajorVersion)
                                           throws UnsupportedKerasConfigurationException,
                                                  InvalidKerasConfigurationException

Map Keras to DL4J weight initialization functions.

Parameters:

kerasInit - String containing Keras initialization function name

Returns:

DL4J weight initialization enum

Throws:

UnsupportedKerasConfigurationException

InvalidKerasConfigurationException

See Also:

WeightInit

getWeightInitFromConfig

public static IWeightInit getWeightInitFromConfig(Map<String,Object> layerConfig,
                                                  String initField,
                                                  boolean enforceTrainingConfig,
                                                  KerasLayerConfiguration conf,
                                                  int kerasMajorVersion)
                                           throws InvalidKerasConfigurationException,
                                                  UnsupportedKerasConfigurationException

Get weight initialization from Keras layer configuration.

Parameters:

layerConfig - dictionary containing Keras layer configuration

enforceTrainingConfig - whether to enforce loading configuration for further training

Returns:

Pair of DL4J weight initialization and distribution

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config