/** * Strips the data transform of all but the passed in labels * * @param labels strips the data transform of all but the passed in labels * @return the dataset with only the specified labels */ @Override public DataSet filterBy(int[] labels) { List<DataSet> list = asList(); List<DataSet> newList = new ArrayList<>(); List<Integer> labelList = new ArrayList<>(); for (int i : labels) labelList.add(i); for (DataSet d : list) { int outcome = d.outcome(); if (labelList.contains(outcome)) { newList.add(d); } } return DataSet.merge(newList); }
for (int i = 0; i < next.numExamples(); i++) { DataSet currExample = next.get(i); if (!labelRootDirs.get(currExample.outcome()).exists()) labelRootDirs.get(currExample.outcome()).mkdirs(); File example = new File(labelRootDirs.get(currExample.outcome()), String.valueOf(paths.get(currExample.outcome()).size())); currExample.save(example); paths.get(currExample.outcome()).add(example);
int examples = numExamples(); for (DataSet d : data) { int label = d.outcome(); Queue<DataSet> q = map.get(label); if (q == null) {
int o2 = example.outcome(); Integer outcome = labelMap.get(o2); newLabels.add(outcome);
/** * Strips the data transform of all but the passed in labels * * @param labels strips the data transform of all but the passed in labels * @return the dataset with only the specified labels */ @Override public DataSet filterBy(int[] labels) { List<DataSet> list = asList(); List<DataSet> newList = new ArrayList<>(); List<Integer> labelList = new ArrayList<>(); for (int i : labels) labelList.add(i); for (DataSet d : list) { int outcome = d.outcome(); if (labelList.contains(outcome)) { newList.add(d); } } return DataSet.merge(newList); }
for (int i = 0; i < next.numExamples(); i++) { DataSet currExample = next.get(i); if (!labelRootDirs.get(currExample.outcome()).exists()) labelRootDirs.get(currExample.outcome()).mkdirs(); File example = new File(labelRootDirs.get(currExample.outcome()), String.valueOf(paths.get(currExample.outcome()).size())); currExample.save(example); paths.get(currExample.outcome()).add(example);
int examples = numExamples(); for (DataSet d : data) { int label = d.outcome(); Queue<DataSet> q = map.get(label); if (q == null) {
int o2 = example.outcome(); Integer outcome = labelMap.get(o2); newLabels.add(outcome);