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

Example 1 with SimpleNaiveBayesClassifier

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

the class ConfusionMatrixGeneratorTest method testGetConfusionMatrixWithSNB.

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