edu.stanford.nlp.classify
Class KNN

java.lang.Object
  |
  +--edu.stanford.nlp.classify.AbstractClassifier
        |
        +--edu.stanford.nlp.classify.KNN

All Implemented Interfaces:

Classifier

public class KNN

extends AbstractClassifier

K nearest-neighbors classifier.

Field Summary

Fields inherited from class edu.stanford.nlp.classify.AbstractClassifier

testmatrix, trainlabels, trainmatrix

Constructor Summary

KNN()

Method Summary

protected int	assignCategory(int datum) returns number of nearest neighbors that were classified as category category
protected int	assignCategory(int datum, Array nearest) assigns datum datum to a category and returns the index of the category.
protected Array	assignMultipleCategories(int datum) Assigns Datum datum to multiple categories with certain probabilities.
protected Array	assignMultipleCategories(int datum, Array nearest) assigns datum datum to multiple categories with certain probabilities.
protected Array	nearestNeighbors(Array newDatum, int k) returns the indices of the k nearest neighbors (in the training set) to a given test Array.
void	setK(int k) sets the number of nearest neighbors to use in K-Nearest-Neighbors algorithm.

Methods inherited from class edu.stanford.nlp.classify.AbstractClassifier

classify, initialize, softClassify

Methods inherited from class java.lang.Object

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

Constructor Detail

KNN

public KNN()

Method Detail

setK

public void setK(int k)

sets the number of nearest neighbors to use in K-Nearest-Neighbors algorithm. Default is 5.

nearestNeighbors

protected Array nearestNeighbors(Array newDatum,
                                 int k)

returns the indices of the k nearest neighbors (in the training set) to a given test Array.

assignCategory

protected int assignCategory(int datum,
                             Array nearest)

assigns datum datum to a category and returns the index of the category.

assignCategory

protected int assignCategory(int datum)

returns number of nearest neighbors that were classified as category category

Specified by:

assignCategory in class AbstractClassifier

assignMultipleCategories

protected Array assignMultipleCategories(int datum,
                                         Array nearest)

assigns datum datum to multiple categories with certain probabilities. returns array of category_index-probability pairs.

assignMultipleCategories

protected Array assignMultipleCategories(int datum)

Description copied from class: AbstractClassifier

Assigns Datum datum to multiple categories with certain probabilities. returns array of category_index-probability pairs.

Specified by:

assignMultipleCategories in class AbstractClassifier