org.deeplearning4j.nn.modelimport.keras.trainedmodels

Class TrainedModelHelper

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.trainedmodels.TrainedModelHelper

public class TrainedModelHelper
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
protected static org.slf4j.Logger	logger

Constructor Summary

Constructors

Constructor and Description
TrainedModelHelper(TrainedModels model)

Method Summary

Modifier and Type	Method and Description
ComputationGraph	loadModel()
void	setPathToH5(java.lang.String pathtoH5)
void	setPathToJSON(java.lang.String pathToJSON)

Methods inherited from class java.lang.Object

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

Field Detail

logger

protected static final org.slf4j.Logger logger

Constructor Detail

TrainedModelHelper

public TrainedModelHelper(TrainedModels model)

Method Detail

setPathToH5

public void setPathToH5(java.lang.String pathtoH5)

setPathToJSON

public void setPathToJSON(java.lang.String pathToJSON)

loadModel

public ComputationGraph loadModel()
                           throws java.io.IOException,
                                  InvalidKerasConfigurationException,
                                  UnsupportedKerasConfigurationException

Throws:

java.io.IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException