public class KerasModelUtils
extends java.lang.Object
Constructor and Description |
---|
KerasModelUtils() |
Modifier and Type | Method and Description |
---|---|
static Model |
copyWeightsToModel(Model model,
java.util.Map<java.lang.String,KerasLayer> kerasLayers)
Helper function to import weights from nested Map into existing model.
|
static java.lang.String |
determineKerasBackend(java.util.Map<java.lang.String,java.lang.Object> modelConfig,
KerasModelConfiguration config)
Determine Keras backend
|
static int |
determineKerasMajorVersion(java.util.Map<java.lang.String,java.lang.Object> modelConfig,
KerasModelConfiguration config)
Determine Keras major version
|
static void |
importWeights(Hdf5Archive weightsArchive,
java.lang.String weightsRoot,
java.util.Map<java.lang.String,KerasLayer> layers,
int kerasVersion,
java.lang.String backend)
Store weights to import with each associated Keras layer.
|
static java.util.Map<java.lang.String,java.lang.Object> |
parseJsonString(java.lang.String json)
Convenience function for parsing JSON strings.
|
static java.util.Map<java.lang.String,java.lang.Object> |
parseModelConfig(java.lang.String modelJson,
java.lang.String modelYaml)
Parse Keras model configuration from JSON or YAML string representation
|
static java.util.Map<java.lang.String,java.lang.Object> |
parseYamlString(java.lang.String yaml)
Convenience function for parsing YAML strings.
|
public static Model copyWeightsToModel(Model model, java.util.Map<java.lang.String,KerasLayer> kerasLayers) throws InvalidKerasConfigurationException
model
- DL4J Model interfaceInvalidKerasConfigurationException
- Invalid Keras configpublic static int determineKerasMajorVersion(java.util.Map<java.lang.String,java.lang.Object> modelConfig, KerasModelConfiguration config) throws InvalidKerasConfigurationException
modelConfig
- parsed model configuration for keras modelconfig
- basic model configuration (KerasModelConfiguration)InvalidKerasConfigurationException
- Invalid Keras configpublic static java.lang.String determineKerasBackend(java.util.Map<java.lang.String,java.lang.Object> modelConfig, KerasModelConfiguration config)
modelConfig
- parsed model configuration for keras modelconfig
- basic model configuration (KerasModelConfiguration)InvalidKerasConfigurationException
- Invalid Keras configpublic 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
weightsArchive
- Hdf5ArchiveweightsRoot
- root of weights in HDF5 archiveInvalidKerasConfigurationException
- Invalid Keras configurationUnsupportedKerasConfigurationException
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
modelJson
- JSON string representing model (potentially null)modelYaml
- YAML string representing model (potentially null)java.io.IOException
- IO exceptionInvalidKerasConfigurationException
- Invalid Keras configpublic static java.util.Map<java.lang.String,java.lang.Object> parseJsonString(java.lang.String json) throws java.io.IOException
json
- String containing valid JSONjava.io.IOException
- IO exceptionpublic static java.util.Map<java.lang.String,java.lang.Object> parseYamlString(java.lang.String yaml) throws java.io.IOException
yaml
- String containing valid YAMLjava.io.IOException
- IO exception