Skip to content

Instantly share code, notes, and snippets.

@confluentgist

confluentgist/load_model.java

Forked from kaiwaehner/load_model.java
Created October 29, 2019 23:41
Show Gist options
  • Save confluentgist/dc58aee178e3ed70f93591e73a4dc035 to your computer and use it in GitHub Desktop.
Save confluentgist/dc58aee178e3ed70f93591e73a4dc035 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