From 3a7c2ac8cca374e14238e958a13be81e03170e4e Mon Sep 17 00:00:00 2001
From: Alessio Stalla <alessiostalla@gmail.com>
Date: Sun, 13 Oct 2019 09:38:38 +0200
Subject: [PATCH] #BAEL-18260 Try and update the nd4j dependency to fix the
 build

---
 deeplearning4j/pom.xml                                     | 7 ++++++-
 .../src/main/java/com/baeldung/logreg/MnistClassifier.java | 2 +-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/deeplearning4j/pom.xml b/deeplearning4j/pom.xml
index 64d365e188..9a381c6967 100644
--- a/deeplearning4j/pom.xml
+++ b/deeplearning4j/pom.xml
@@ -14,6 +14,11 @@
     </parent>
 
     <dependencies>
+        <dependency>
+            <groupId>org.nd4j</groupId>
+            <artifactId>nd4j-api</artifactId>
+            <version>${dl4j.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.nd4j</groupId>
             <artifactId>nd4j-native-platform</artifactId>
@@ -43,7 +48,7 @@
     </dependencies>
 
     <properties>
-        <dl4j.version>0.9.1</dl4j.version>
+        <dl4j.version>1.0.0-beta5</dl4j.version>
     </properties>
 
 </project>
diff --git a/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java b/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java
index 1246de973f..e320c366bf 100644
--- a/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java
+++ b/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java
@@ -165,4 +165,4 @@ public class MnistClassifier {
         ModelSerializer.writeModel(model, modelPath, true);
         logger.info("The MINIST model has been saved in {}", modelPath.getPath());
     }
-}
\ No newline at end of file
+}