tech.tablesaw.api.ml.classification

Class RandomForest

java.lang.Object

tech.tablesaw.api.ml.classification.AbstractClassifier

tech.tablesaw.api.ml.classification.RandomForest

public class RandomForest
extends AbstractClassifier

Method Summary

Modifier and Type	Method and Description
static RandomForest	learn(int nTrees, CategoryColumn classes, NumericColumn... columns)
static RandomForest	learn(int nTrees, IntColumn classes, NumericColumn... columns)
static RandomForest	learn(int nTrees, 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 RandomForest learn(int nTrees,
                                 IntColumn classes,
                                 NumericColumn... columns)

learn

public static RandomForest learn(int nTrees,
                                 ShortColumn classes,
                                 NumericColumn... columns)

learn

public static RandomForest 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)