Examples with KNearestNeighborClassifier org.apache.lucene.classification.KNearestNeighborClassifier used on opensource projects

Example 1 with KNearestNeighborClassifier

use of org.apache.lucene.classification.KNearestNeighborClassifier in project lucene-solr by apache.

the class ConfusionMatrixGeneratorTest method testGetConfusionMatrixWithKNN.

@Test
public void testGetConfusionMatrixWithKNN() throws Exception {
    LeafReader reader = null;
    try {
        MockAnalyzer analyzer = new MockAnalyzer(random());
        reader = getSampleIndex(analyzer);
        Classifier<BytesRef> classifier = new KNearestNeighborClassifier(reader, null, analyzer, null, 1, 0, 0, categoryFieldName, textFieldName);
        ConfusionMatrixGenerator.ConfusionMatrix confusionMatrix = ConfusionMatrixGenerator.getConfusionMatrix(reader, classifier, categoryFieldName, textFieldName, -1);
        checkCM(confusionMatrix);
    } finally {
        if (reader != null) {
            reader.close();
        }
    }
}