Make PartionedDataSet store LearningDataSets to make use of the LearningDataSet APIs

Author: Zomis

src/main/java/net/zomis/machlearn/common/LearningDataSet.java

@@ -12,7 +12,15 @@
 
 public class LearningDataSet {
 
-    private final List<LearningData> data = new ArrayList<>();
+    private final List<LearningData> data;
+
+    public LearningDataSet() {
+        this(new ArrayList<>());
+    }
+
+    public LearningDataSet(List<LearningData> data) {
+        this.data = new ArrayList<>(data);
+    }
 
     public void add(Object representation, double[] x, double y) {
         this.add(representation, x, new double[]{y});

src/main/java/net/zomis/machlearn/common/PartitionedDataSet.java

@@ -2,35 +2,34 @@
 
 import net.zomis.machlearn.neural.LearningData;
 
-import java.util.ArrayList;
 import java.util.List;
 
 /**
  * Represents a LearningDataSet that has been partitioned into training set, cross-validation set, and/or test-set.
  */
 public class PartitionedDataSet {
 
-    private final List<LearningData> trainingSet;
-    private final List<LearningData> crossValidationSet;
-    private final List<LearningData> testSet;
+    private final LearningDataSet trainingSet;
+    private final LearningDataSet crossValidationSet;
+    private final LearningDataSet testSet;
 
     public PartitionedDataSet(List<LearningData> trainingSet, List<LearningData> crossValidationSet,
                               List<LearningData> testSet) {
-        this.trainingSet = new ArrayList<>(trainingSet);
-        this.crossValidationSet = new ArrayList<>(crossValidationSet);
-        this.testSet = new ArrayList<>(testSet);
+        this.trainingSet = new LearningDataSet(trainingSet);
+        this.crossValidationSet = new LearningDataSet(crossValidationSet);
+        this.testSet = new LearningDataSet(testSet);
     }
 
-    public List<LearningData> getCrossValidationSet() {
-        return new ArrayList<>(crossValidationSet);
+    public LearningDataSet getCrossValidationSet() {
+        return crossValidationSet;
     }
 
-    public List<LearningData> getTestSet() {
-        return new ArrayList<>(testSet);
+    public LearningDataSet getTestSet() {
+        return testSet;
     }
 
-    public List<LearningData> getTrainingSet() {
-        return new ArrayList<>(trainingSet);
+    public LearningDataSet getTrainingSet() {
+        return trainingSet;
     }
 
 }