de.edux.util.LabelDimensionConverter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of lib Show documentation
Show all versions of lib Show documentation
Educational library for machine learning challenges
The newest version!
package de.edux.util;
public class LabelDimensionConverter {
public static int[] convert2DLabelArrayTo1DLabelArray(double[][] labels) {
int[] decisionTreeTrainLabels = new int[labels.length];
for (int i = 0; i < labels.length; i++) {
for (int j = 0; j < labels[i].length; j++) {
if (labels[i][j] == 1) {
decisionTreeTrainLabels[i] = (j + 1);
}
}
}
return decisionTreeTrainLabels;
}
}