package cn.edu.bjut.textmining.chapter4;

import cn.edu.bjut.textmining.chapter4.support.EvaluationSupport;
import cn.edu.bjut.textmining.chapter4.support.NewsClassifier;
import cn.edu.bjut.textmining.chapter4.support.NewsExperiment;

/** OpenNLP朴素贝叶斯：中文词元到类别得分。 */
public final class OpenNlpNaiveBayesDemo {
    public static void main(String[] args) {
        try {
            NewsExperiment experiment = NewsExperiment.load();
            experiment.printSummary();
            NewsClassifier model = new OpenNlpNewsClassifier();
            model.fit(experiment.training);
            EvaluationSupport.showExamples(model, experiment);
            EvaluationSupport.evaluate(model, experiment);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}