tech.tablesaw.api.ml.classification

Class DecisionTree

java.lang.Object

tech.tablesaw.api.ml.classification.AbstractClassifier

tech.tablesaw.api.ml.classification.DecisionTree

public class DecisionTree
extends AbstractClassifier

Method Summary

Modifier and Type	Method and Description
static DecisionTree	learn(int nTrees, CategoryColumn classes, NumericColumn... columns)
static DecisionTree	learn(int maxNodes, IntColumn classes, NumericColumn... columns)
static DecisionTree	learn(int maxNodes, ShortColumn classes, NumericColumn... columns)
int	predict(double[] data)
ConfusionMatrix	predictMatrix(CategoryColumn labels, NumericColumn... predictors)
ConfusionMatrix	predictMatrix(ShortColumn labels, NumericColumn... predictors)

Methods inherited from class java.lang.Object

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

Method Detail

learn

public static DecisionTree learn(int maxNodes,
                                 IntColumn classes,
                                 NumericColumn... columns)

learn

public static DecisionTree learn(int maxNodes,
                                 ShortColumn classes,
                                 NumericColumn... columns)

learn

public static DecisionTree learn(int nTrees,
                                 CategoryColumn classes,
                                 NumericColumn... columns)

predict

public int predict(double[] data)

predictMatrix

public ConfusionMatrix predictMatrix(ShortColumn labels,
                                     NumericColumn... predictors)

predictMatrix

public ConfusionMatrix predictMatrix(CategoryColumn labels,
                                     NumericColumn... predictors)