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;

/** Weka kNN：训练集拟合TF-IDF，然后以k=5近邻分类。 */
public final class KnnWekaDemo {
    public static void main(String[] args) {
        try {
            NewsExperiment experiment = NewsExperiment.load();
            experiment.printSummary();
            NewsClassifier model = new WekaNewsClassifier();
            model.fit(experiment.training);
            EvaluationSupport.showExamples(model, experiment);
            EvaluationSupport.evaluate(model, experiment);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}