Skip to content

Instantly share code, notes, and snippets.

@kaiwaehner

kaiwaehner/load_model.java

Created October 29, 2019 12:16
Show Gist options
  • Save kaiwaehner/30e92ecc0598fa630f6733b98b9dae8c to your computer and use it in GitHub Desktop.
Save kaiwaehner/30e92ecc0598fa630f6733b98b9dae8c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_model.java
// Step 1: Load Keras TensorFlow Model using DeepLearning4J API
String simpleMlp = new ClassPathResource("generatedModels/Keras/simple_mlp.h5").getFile().getPath();
System.out.println(simpleMlp.toString());
MultiLayerNetwork model = KerasModelImport.importKerasSequentialModelAndWeights(simpleMlp);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment