| Package | Description |
|---|---|
| org.deeplearning4j.nn.modelimport.keras | |
| org.deeplearning4j.nn.modelimport.keras.utils |
| Modifier and Type | Field and Description |
|---|---|
protected KerasModelBuilder |
KerasModel.modelBuilder |
| Modifier and Type | Method and Description |
|---|---|
KerasModelBuilder |
KerasModel.modelBuilder() |
| Constructor and Description |
|---|
KerasModel(KerasModelBuilder modelBuilder)
(Recommended) Builder-pattern constructor for (Functional API) Model.
|
KerasSequentialModel(KerasModelBuilder modelBuilder)
(Recommended) Builder-pattern constructor for Sequential model.
|
| Modifier and Type | Method and Description |
|---|---|
KerasModelBuilder |
KerasModelBuilder.dimOrder(KerasLayer.DimOrder dimOrder)
Manually set dim order for Keras model, i.e.
|
KerasModelBuilder |
KerasModelBuilder.enforceTrainingConfig(boolean enforceTrainingConfig)
Determine whether to enforce loading a training configuration or not.
|
KerasModelBuilder |
KerasModelBuilder.inputShape(int[] inputShape)
Provide input shape for Keras models that have been compiled without one.
|
KerasModelBuilder |
KerasModelBuilder.modelHdf5Filename(String modelHdf5Filename)
Set full model HDF5 (architecture, weights and training configuration) by providing the HDF5 filename.
|
KerasModelBuilder |
KerasModelBuilder.modelJson(String modelJson)
Set model architecture from model JSON string.
|
KerasModelBuilder |
KerasModelBuilder.modelJsonFilename(String modelJsonFilename)
Set model architecture from file name pointing to model JSON string.
|
KerasModelBuilder |
KerasModelBuilder.modelJsonInputStream(InputStream modelJsonInputStream)
Set model architecture from input stream of model JSON.
|
KerasModelBuilder |
KerasModelBuilder.modelYaml(String modelYaml)
Set model architecture from model YAML string.
|
KerasModelBuilder |
KerasModelBuilder.modelYamlFilename(String modelYamlFilename)
Set model architecture from file name pointing to model YAML string.
|
KerasModelBuilder |
KerasModelBuilder.modelYamlInputStream(InputStream modelYamlInputStream)
Set model architecture from input stream of model YAML.
|
KerasModelBuilder |
KerasModelBuilder.trainingJson(String trainingJson)
Provide training configuration as JSON string
|
KerasModelBuilder |
KerasModelBuilder.trainingJsonInputStream(InputStream trainingJsonInputStream)
Provide training configuration as file input stream from JSON
|
KerasModelBuilder |
KerasModelBuilder.trainingYaml(String trainingYaml)
Provide training configuration as YAML string
|
KerasModelBuilder |
KerasModelBuilder.trainingYamlInputStream(InputStream trainingYamlInputStream)
Provide training configuration as file input stream from YAML
|
KerasModelBuilder |
KerasModelBuilder.weightsHdf5Filename(String weightsHdf5Filename)
Set weights of the model by providing the file name of the corresponding weights HDF5 file.
|
KerasModelBuilder |
KerasModelBuilder.weightsHdf5FilenameNoRoot(String weightsHdf5Filename)
Set weights of the model by providing the file name of the corresponding weights HDF5 file.
|
Copyright © 2021. All rights reserved.