weka classifiers NaiveBayes - Java Machine Learning AI

Java examples for Machine Learning AI:weka

  1. HOME
  2. Java
  3. Machine Learning AI
  4. weka

Description

weka classifiers NaiveBayes

Demo Code



import java.io.BufferedReader;
import java.io.FileReader;
import weka.classifiers.evaluation.output.prediction.PlainText;

import weka.classifiers.Evaluation;
import weka.core.Instances;

public class NaiveBayes {

    public static void main(String[] args) throws Exception {
        BufferedReader readerTrain = new BufferedReader(new FileReader(
                args[0]));//from w  ww  .  j a  va 2s  .  co m
        BufferedReader readerTest = new BufferedReader(new FileReader(
                args[1]));
        Instances train = new Instances(readerTrain);
        Instances test = new Instances(readerTest);
        train.setClassIndex(0);
        test.setClassIndex(0);
        weka.classifiers.bayes.NaiveBayes cls = new weka.classifiers.bayes.NaiveBayes();
        cls.buildClassifier(train);
        Evaluation eval = new Evaluation(train);
        eval.evaluateModel(cls, test);
        System.out.println(eval
                .toSummaryString("\nResults\n======\n", true));
        System.out.println(eval.toClassDetailsString());
        readerTrain.close();
        readerTest.close();
    }

}

Related Tutorials