org.apache.mahout.df.builder
Class DefaultTreeBuilder
java.lang.Object
org.apache.mahout.df.builder.DefaultTreeBuilder
- All Implemented Interfaces:
- TreeBuilder
public class DefaultTreeBuilder
- extends java.lang.Object
- implements TreeBuilder
Builds a Decision Tree
Based on the algorithm described in the "Decision Trees" tutorials by Andrew W. Moore, available at:
http://www.cs.cmu.edu/~awm/tutorials
Method Summary |
Node |
build(java.util.Random rng,
Data data)
Builds a Decision tree using the training data |
protected static int[] |
randomAttributes(java.util.Random rng,
boolean[] selected,
int m)
Randomly selects m attributes to consider for split, excludes IGNORED and LABEL attributes |
void |
setIgSplit(IgSplit igSplit)
|
void |
setM(int m)
|
Methods inherited from class java.lang.Object |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
DefaultTreeBuilder
public DefaultTreeBuilder()
setM
public void setM(int m)
setIgSplit
public void setIgSplit(IgSplit igSplit)
build
public Node build(java.util.Random rng,
Data data)
- Description copied from interface:
TreeBuilder
- Builds a Decision tree using the training data
- Specified by:
build
in interface TreeBuilder
- Parameters:
rng
- random-numbers generatordata
- training data
- Returns:
- root Node
randomAttributes
protected static int[] randomAttributes(java.util.Random rng,
boolean[] selected,
int m)
- Randomly selects m attributes to consider for split, excludes IGNORED and LABEL attributes
- Parameters:
rng
- random-numbers generatorselected
- attributes' state (selected or not)m
- number of attributes to choose
Copyright © 2008-2010 The Apache Software Foundation. All Rights Reserved.