Example usage for org.apache.lucene.classification SimpleNaiveBayesClassifier SimpleNaiveBayesClassifier

Introduction

In this page you can find the example usage for org.apache.lucene.classification SimpleNaiveBayesClassifier SimpleNaiveBayesClassifier.

Prototype

SimpleNaiveBayesClassifier

Source Link

Usage

From source file:de.uni_koeln.spinfo.textengineering.tm.classification.lucene.TestLuceneClassifier.java

License:Open Source License

private void testWith(final String query) throws IOException {
    setupData(query);/*from w  w  w .j  ava2  s .  c  om*/
    printInfo(query);

    // unser eigener Classifier:
    testEval(System.nanoTime(), query, new NaiveBayes());
    System.out.println(LINE);

    // und Lucene: der LuceneAdapter delegiert die Aufgabe an die Lucene-Api:
    testEval(System.nanoTime(), query, new LuceneAdapter(new SimpleNaiveBayesClassifier(), indexDir, query));
    System.out.println(LINE);

    // und Lucene: der LuceneAdapter delegiert die Aufgabe an die Lucene-Api:
    testEval(System.nanoTime(), query, new LuceneAdapter(new KNearestNeighborClassifier(1), indexDir, query));
    System.out.println(LINE);

}

---

From source file:org.solr.classtify.SimpleNaiveBayesClassifierTest.java

License:Apache License

@Test
public void classtify() throws IOException {
    SimpleNaiveBayesClassifier classifier = new SimpleNaiveBayesClassifier();
    IndexReader reader = DirectoryReader.open(FSDirectory.open(new File(dir)));

    AtomicReader wrap = SlowCompositeReaderWrapper.wrap(reader);
    classifier.train(wrap, textFieldName, categoryFieldName, analyzer);
    ClassificationResult<BytesRef> assignClass = classifier.assignClass(newText);
    BytesRef assignedClass = assignClass.getAssignedClass();

    double score = assignClass.getScore();
    System.out.println(assignedClass.utf8ToString() + "," + score);
}

---