Merge pull request #24 from dmi3coder/feature/saveKernelTypoFix

Feature/save kernel typo fix

Author: dmi3coder

build.gradle

@@ -22,7 +22,7 @@ publishing {
         maven(MavenPublication) {
             groupId = 'de.dmi3y.behaiv'
             artifactId = 'behaiv'
-            version = '0.3.9-alpha'
+            version = '0.4.7-alpha'
             from components.java
         }
     }

src/main/java/de/dmi3y/behaiv/kernel/LogisticRegressionKernel.java

@@ -114,9 +114,12 @@ public void save(BehaivStorage storage) throws IOException {
         if (theta == null && (data == null || data.isEmpty())) {
             throw new IOException("Not enough data to save, network data is empty");
         }
-        if (labels == null || labels.isEmpty()) {
+        if(labels == null || labels.isEmpty()) {
+            labels = toDistinctListOfPairValues(data);
+        }
+        if (labels.isEmpty()) {
             String message;
-            message = "Kernel collected labels but failed to get data, couldn't save network.";
+            message = "Kernel collected data but failed to get labels, couldn't save network.";
             throw new IOException(message);
         }
         if (theta == null) {

src/test/java/de/dmi3y/behaiv/kernel/LogisticRegressionKernelTest.java

@@ -138,17 +138,6 @@ public void storeResults_saveWhenDataIsNull_expectException() throws IOException
         }
     }
 
-    @Test
-    public void storeResults_saveWhenLabelsIsNull_expectException() throws IOException, ClassNotFoundException {
-        LogisticRegressionKernel kernel = new LogisticRegressionKernel("storeTest");
-        kernel.data = getTrainingData();
-        try {
-            kernel.save(storage);
-            fail();
-        } catch (IOException e) {
-            assertEquals(e.getMessage(), "Kernel collected labels but failed to get data, couldn't save network.");
-        }
-    }
 
 
     @Test