Package org.apache.mahout.classifier.df.node

Examples of org.apache.mahout.classifier.df.node.Node

89
90
91
92
93
94
95
96
97
98
99
  @Test
  public void testPredictTrace() throws Exception {
    // build tree
    DecisionTreeBuilder builder = new DecisionTreeBuilder();
    builder.setM(data.getDataset().nbAttributes() - 1);
    Node tree = builder.build(rng, data);
   
    String[] prediction = TreeVisualizer.predictTrace(tree, testData, ATTR_NAMES);
    Assert.assertArrayEquals(prediction, new String[] {
      "outlook = rainy -> windy = TRUE -> no", "outlook = overcast -> yes",
      "outlook = sunny -> (humidity = 90) >= 85 -> no"});
View Full Code Here
128
129
130
131
132
133
134
135
136
137
138
      partitions.add(partition);

      int nbTrees = Step1Mapper.nbTrees(NUM_MAPS, NUM_TREES, partition);

      for (int treeId = 0; treeId < nbTrees; treeId++) {
        Node tree = new Leaf(rng.nextInt(100));

        keys[index] = new TreeID(partition, treeId);
        values[index] = new MapredOutput(tree, nextIntArray(rng, NUM_INSTANCES));

        index++;
View Full Code Here
149
150
151
152
153
154
155
156
157
158
159
   
    log.debug("Building {} trees", nbTrees);
    for (int treeId = 0; treeId < nbTrees; treeId++) {
      log.debug("Building tree number : {}", treeId);
     
      Node tree = bagging.build(rng);
     
      key.set(partition, firstTreeId + treeId);
     
      if (isOutput()) {
        MapredOutput emOut = new MapredOutput(tree);
View Full Code Here
81
82
83
84
85
86
87
88
89
90
91
  @Test
  public void testTreeVisualize() throws Exception {
    // build tree
    DecisionTreeBuilder builder = new DecisionTreeBuilder();
    builder.setM(data.getDataset().nbAttributes() - 1);
    Node tree = builder.build(rng, data);

    String visualization = TreeVisualizer.toString(tree, data.getDataset(), ATTR_NAMES);

    assertTrue(
        ("\n" +
View Full Code Here
109
110
111
112
113
114
115
116
117
118
119
  @Test
  public void testPredictTrace() throws Exception {
    // build tree
    DecisionTreeBuilder builder = new DecisionTreeBuilder();
    builder.setM(data.getDataset().nbAttributes() - 1);
    Node tree = builder.build(rng, data);
   
    String[] prediction = TreeVisualizer.predictTrace(tree, testData,
        ATTR_NAMES);
    Assert.assertArrayEquals(new String[] {
        "outlook = rainy -> windy = TRUE -> no", "outlook = overcast -> yes",
View Full Code Here
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
  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

Related Classes of org.apache.mahout.classifier.df.node.Node

  • org.apache.mahout.classifier.df.builder.DecisionTreeBuilder
  • org.apache.mahout.classifier.df.builder.DefaultTreeBuilder
  • org.apache.mahout.classifier.df.mapreduce.inmem.InMemMapper
  • org.apache.mahout.classifier.df.mapreduce.partial.PartialBuilderTest
  • org.apache.mahout.classifier.df.mapreduce.partial.Step1Mapper
  • org.apache.mahout.classifier.df.tools.TreeVisualizer
  • org.apache.mahout.classifier.df.tools.VisualizerTest
    • Copyright © 2018 www.massapicom. 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.