target column names fix

Author: deepnetts

Diff for: src/main/java/jsr381/example/util/DataSetExamples.java

@@ -22,8 +22,10 @@
 import java.nio.file.Paths;
 import java.util.Arrays;
 import java.util.Enumeration;
+import java.util.List;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipFile;
+import javax.visrec.ml.data.Column;
 
 /**
  * @author Kevin Berendsen
@@ -77,6 +79,11 @@ private static BasicDataSet fromURL(URL url, String delimiter, int inputsNum, in
             dataSet.setColumnNames(colNames);
         }
 
+        List<Column> cols = dataSet.getColumns();
+        for(int i=0; i<outputsNum; i++) {
+            //dataSet.getColumns().get(inputsNum+i).setAsTarget(true);          
+          cols.get(inputsNum+i).setAsTarget(true);
+        }
 
         Arrays.stream(content)
                 .skip(skipCount)
@@ -160,10 +167,6 @@ public static ExampleDataSet getMnistDataSet() throws IOException { // print out
                 .setTrainingFile(trainingIndexFile);
     }
 
-    public static void main(String[] args) throws IOException {
-        System.out.println(getMnistDataSet());
-    }
-
     public static class ExampleDataSet {
         private File labelsFile;
         private File trainingFile;