jsat.classifiers.trees

Class TreePruner

java.lang.Object

jsat.classifiers.trees.TreePruner

public class TreePruner
extends Object

Provides post-pruning algorithms for any decision tree that can be altered using the TreeNodeVisitor. Pruning is done with a held out testing set
All pruning methods handle missing values
NOTE: API still under work, expect changes

Author:

Edward Raff

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TreePruner.PruningMethod The method of pruning to use

Method Summary

Modifier and Type	Method and Description
static void	prune(TreeNodeVisitor root, TreePruner.PruningMethod method, ClassificationDataSet testSet) Performs pruning starting from the root node of a tree
static void	prune(TreeNodeVisitor root, TreePruner.PruningMethod method, List<DataPointPair<Integer>> testSet) Performs pruning starting from the root node of a tree

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

prune

public static void prune(TreeNodeVisitor root,
                         TreePruner.PruningMethod method,
                         ClassificationDataSet testSet)

Performs pruning starting from the root node of a tree

Parameters:

root - the root node of a decision tree

method - the pruning method to use

testSet - the test set of data points to use for pruning

prune

public static void prune(TreeNodeVisitor root,
                         TreePruner.PruningMethod method,
                         List<DataPointPair<Integer>> testSet)

Performs pruning starting from the root node of a tree

Parameters:

root - the root node of a decision tree

method - the pruning method to use

testSet - the test set of data points to use for pruning