@Override public DecisionTree train(double[][] x, int[] y) { return new DecisionTree(attributes, x, y, maxNodes, nodeSize, rule); } }
private DecisionTree(int maxNodes, int[] classArray, NumericColumn... columns) { double[][] data = DoubleArrays.to2dArray(columns); this.classifierModel = new smile.classification.DecisionTree(data, classArray, maxNodes); }
trees[t] = new DecisionTree(attributes, x, y, maxNodes, 1, x[0].length, DecisionTree.SplitRule.GINI, samples, order);
DecisionTree tree = new DecisionTree(attributes, x, y, maxNodes, nodeSize, mtry, rule, samples.clone(), order);