Package org.apache.mahout.classifier.df.builder

Examples of org.apache.mahout.classifier.df.builder.DecisionTreeBuilder.build()

182
183
184
185
186
187
188
189
190
191
192
    // build tree
    DecisionTreeBuilder builder = new DecisionTreeBuilder();
    builder.setM(data.getDataset().nbAttributes() - 1);
    builder.setMinSplitNum(0);
    builder.setComplemented(false);
    Node tree = builder.build(rng, lessData);

    String visualization = TreeVisualizer.toString(tree, data.getDataset(), ATTR_NAMES);
    assertTrue(
        ("\noutlook = sunny\n" +
            "|   humidity < 77.5 : yes\n" +
View Full Code Here
202
203
204
205
206
207
208
209
210
211
  public void testEmpty() throws Exception {
    Data emptyData = new Data(data.getDataset());
   
    // build tree
    DecisionTreeBuilder builder = new DecisionTreeBuilder();
    Node tree = builder.build(rng, emptyData);

    assertEquals(" : unknown", TreeVisualizer.toString(tree, data.getDataset(), ATTR_NAMES));
  }
}
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.