tech.tablesaw.api.ml.classification

Class Knn

java.lang.Object

tech.tablesaw.api.ml.classification.AbstractClassifier

tech.tablesaw.api.ml.classification.Knn

public class Knn
extends AbstractClassifier

Method Summary

Modifier and Type	Method and Description
static Knn	learn(int k, BooleanColumn labels, NumericColumn... predictors)
static Knn	learn(int k, CategoryColumn labels, NumericColumn... predictors)
static Knn	learn(int k, IntColumn labels, NumericColumn... predictors)
static Knn	learn(int k, ShortColumn labels, NumericColumn... predictors)
int	predict(double[] data)
int[]	predict(NumericColumn... predictors)
ConfusionMatrix	predictMatrix(BooleanColumn labels, NumericColumn... predictors)
ConfusionMatrix	predictMatrix(CategoryColumn labels, NumericColumn... predictors)
ConfusionMatrix	predictMatrix(IntColumn 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 Knn learn(int k,
                        ShortColumn labels,
                        NumericColumn... predictors)

learn

public static Knn learn(int k,
                        IntColumn labels,
                        NumericColumn... predictors)

learn

public static Knn learn(int k,
                        BooleanColumn labels,
                        NumericColumn... predictors)

learn

public static Knn learn(int k,
                        CategoryColumn labels,
                        NumericColumn... predictors)

predict

public int predict(double[] data)

predictMatrix

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

predictMatrix

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

predictMatrix

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

predictMatrix

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

predict

public int[] predict(NumericColumn... predictors)