public static void testManualDiscretize() { Node x = new ContinuousVariable("X"); List<Node> nodes = Collections.singletonList(x); DataSet data = new ColtDataSet(9, nodes); data.setDouble(0, 0, 13.0); data.setDouble(1, 0, 1.2); data.setDouble(2, 0, 2.2); data.setDouble(3, 0, 4.5); data.setDouble(4, 0, 12.005); data.setDouble(5, 0, 5.5); data.setDouble(6, 0, 10.1); data.setDouble(7, 0, 7.5); data.setDouble(8, 0, 3.4); System.out.println(data); Discretizer discretizer = new Discretizer(data); discretizer.setVariablesCopied(true); discretizer.equalCounts(x, 3); DataSet discretized = discretizer.discretize(); System.out.println(discretized); assertEquals(discretized.getInt(0, 0), 2); assertEquals(discretized.getInt(1, 0), 0); assertEquals(discretized.getInt(2, 0), 0); assertEquals(discretized.getInt(3, 0), 1); assertEquals(discretized.getInt(4, 0), 2); assertEquals(discretized.getInt(5, 0), 1); assertEquals(discretized.getInt(6, 0), 2); assertEquals(discretized.getInt(7, 0), 1); assertEquals(discretized.getInt(8, 0), 0); }
public void testManualDiscretize3() { Graph graph = new Dag(GraphUtils.randomGraph(5, 0, 5, 3, 3, 3, false)); SemPm pm = new SemPm(graph); SemIm im = new SemIm(pm); DataSet data = im.simulateData(100, false); List<Node> nodes = data.getVariables(); Discretizer discretizer = new Discretizer(data); discretizer.setVariablesCopied(true); discretizer.setVariablesCopied(true); discretizer.equalCounts(nodes.get(0), 3); DataSet discretized = discretizer.discretize(); System.out.println(discretized); assertTrue(discretized.getVariable(0) instanceof DiscreteVariable); assertTrue(discretized.getVariable(1) instanceof ContinuousVariable); assertTrue(discretized.getVariable(2) instanceof ContinuousVariable); assertTrue(discretized.getVariable(3) instanceof ContinuousVariable); assertTrue(discretized.getVariable(4) instanceof ContinuousVariable); }