Constructor Detail

• KerasModelUtils

  public KerasModelUtils()

Method Detail

• copyWeightsToModel

  public static Model copyWeightsToModel(Model model,
                                         java.util.Map<java.lang.String,KerasLayer> kerasLayers)
                                  throws InvalidKerasConfigurationException

  Helper function to import weights from nested Map into existing model. Depends critically on matched layer and parameter names. In general this seems to be straightforward for most Keras models and layersOrdered, but there may be edge cases.

  Parameters:

  model - DL4J Model interface

  Returns:

  DL4J Model interface

  Throws:

  InvalidKerasConfigurationException - Invalid Keras config

• determineKerasMajorVersion

  public static int determineKerasMajorVersion(java.util.Map<java.lang.String,java.lang.Object> modelConfig,
                                               KerasModelConfiguration config)
                                        throws InvalidKerasConfigurationException

  Determine Keras major version

  Parameters:

  modelConfig - parsed model configuration for keras model

  config - basic model configuration (KerasModelConfiguration)

  Returns:

  Major Keras version (1 or 2)

  Throws:

  InvalidKerasConfigurationException - Invalid Keras config

• determineKerasBackend

  public static java.lang.String determineKerasBackend(java.util.Map<java.lang.String,java.lang.Object> modelConfig,
                                                       KerasModelConfiguration config)

  Determine Keras backend

  Parameters:

  modelConfig - parsed model configuration for keras model

  config - basic model configuration (KerasModelConfiguration)

  Returns:

  Keras backend string

  Throws:

  InvalidKerasConfigurationException - Invalid Keras config

• importWeights

  public static void importWeights(Hdf5Archive weightsArchive,
                                   java.lang.String weightsRoot,
                                   java.util.Map<java.lang.String,KerasLayer> layers,
                                   int kerasVersion,
                                   java.lang.String backend)
                            throws InvalidKerasConfigurationException,
                                   UnsupportedKerasConfigurationException

  Store weights to import with each associated Keras layer.

  Parameters:

  weightsArchive - Hdf5Archive

  weightsRoot - root of weights in HDF5 archive

  Throws:

  InvalidKerasConfigurationException - Invalid Keras configuration

  UnsupportedKerasConfigurationException

• parseModelConfig

  public static java.util.Map<java.lang.String,java.lang.Object> parseModelConfig(java.lang.String modelJson,
                                                                                  java.lang.String modelYaml)
                                                                           throws java.io.IOException,
                                                                                  InvalidKerasConfigurationException

  Parse Keras model configuration from JSON or YAML string representation

  Parameters:

  modelJson - JSON string representing model (potentially null)

  modelYaml - YAML string representing model (potentially null)

  Returns:

  Model configuration as Map

  Throws:

  java.io.IOException - IO exception

  InvalidKerasConfigurationException - Invalid Keras config

• parseJsonString

  public static java.util.Map<java.lang.String,java.lang.Object> parseJsonString(java.lang.String json)
                                                                          throws java.io.IOException

  Convenience function for parsing JSON strings.

  Parameters:

  json - String containing valid JSON

  Returns:

  Nested (key,value) map of arbitrary depth

  Throws:

  java.io.IOException - IO exception

• parseYamlString

  public static java.util.Map<java.lang.String,java.lang.Object> parseYamlString(java.lang.String yaml)
                                                                          throws java.io.IOException

  Convenience function for parsing YAML strings.

  Parameters:

  yaml - String containing valid YAML

  Returns:

  Nested (key,value) map of arbitrary depth

  Throws:

  java.io.IOException - IO exception