@Override public void shuffle() { List<org.nd4j.linalg.dataset.api.MultiDataSet> split = asList(); Collections.shuffle(split); MultiDataSet mds = merge(split); this.features = mds.features; this.labels = mds.labels; this.featuresMaskArrays = mds.featuresMaskArrays; this.labelsMaskArrays = mds.labelsMaskArrays; this.exampleMetaData = mds.exampleMetaData; }
@Override public MultiDataSet next(int num) { int end = curr + num; List<MultiDataSet> r = new ArrayList<>(); if (end >= list.size()) { end = list.size(); } for (; curr < end; curr++) { r.add(list.get(curr)); } MultiDataSet d = org.nd4j.linalg.dataset.MultiDataSet.merge(r); if (preProcessor != null) { preProcessor.preProcess(d); } return d; }
@Override public MultiDataSet merge(List<MultiDataSet> toMerge) { return org.nd4j.linalg.dataset.MultiDataSet.merge(toMerge); }
@Override public MultiDataSet next(int num) { int end = curr + num; List<MultiDataSet> r = new ArrayList<>(); if (end >= list.size()) { end = list.size(); } for (; curr < end; curr++) { r.add(list.get(curr)); } MultiDataSet d = org.nd4j.linalg.dataset.MultiDataSet.merge(r); if (preProcessor != null) { preProcessor.preProcess(d); } return d; }
out = list.get(0); } else { out = org.nd4j.linalg.dataset.MultiDataSet.merge(list);
out = list.get(0); } else { out = org.nd4j.linalg.dataset.MultiDataSet.merge(list);