Use KerasModelImport to load an external MultiLayerNetwork model:
String modelFileLocation = new ClassPathResource("kerasModel.h5").getFile().getPath(); MultiLayerNetwork model = KerasModelImport.importKerasSequentialModelAndWeights(modelFileLocation);
Use KerasModelImport to load an external ComputationGraph model:
String modelFileLocation = new ClassPathResource("kerasModel.h5").getFile().getPath(); ComputationGraph model = KerasModelImport.importKerasModelAndWeights(modelFileLocation);
Use KerasModelBuilder to import an external model:
KerasModelBuilder builder = new KerasModel().modelBuilder().modelHdf5Filename(modelFile.getAbsolutePath()) .enforceTrainingConfig(trainConfigToEnforceOrNot); if (inputShape != null) { builder.inputShape(inputShape); } KerasModel model = builder.buildModel(); ComputationGraph newModel = model.getComputationGraph();