public static void main(String[] args) throws IOException { String inputFile = args[1]; BufferedReader in = new BufferedReader(new FileReader(inputFile)); List<String> symbols = new ArrayList<String>(); String line = in.readLine(); while (line != null) { String[] pieces = line.split(","); if (!symbols.contains(pieces[0])) { symbols.add(pieces[0]); } line = in.readLine(); } ConfusionMatrix x2 = new ConfusionMatrix(symbols, "unknown"); in = new BufferedReader(new FileReader(inputFile)); line = in.readLine(); while (line != null) { String[] pieces = line.split(","); String trueValue = pieces[0]; String estimatedValue = pieces[1]; x2.addInstance(trueValue, estimatedValue); line = in.readLine(); } System.out.printf("%s\n\n", x2.toString()); } }
private static void checkValues(ConfusionMatrix cm) { int[][] counts = cm.getConfusionMatrix(); cm.toString(); assertEquals(counts.length, counts[0].length); assertEquals(3, counts.length); assertEquals(VALUES[0][0], counts[0][0]); assertEquals(VALUES[0][1], counts[0][1]); assertEquals(VALUES[1][0], counts[1][0]); assertEquals(VALUES[1][1], counts[1][1]); assertTrue(Arrays.equals(new int[3], counts[2])); // zeros assertEquals(OTHER[0], counts[0][2]); assertEquals(OTHER[1], counts[1][2]); assertEquals(3, cm.getLabels().size()); assertTrue(cm.getLabels().contains(LABELS[0])); assertTrue(cm.getLabels().contains(LABELS[1])); assertTrue(cm.getLabels().contains(DEFAULT_LABEL)); }