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

Example 1 with CachingNaiveBayesClassifier

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

the class ConfusionMatrixGeneratorTest method testGetConfusionMatrixWithCNB.

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