朴素贝叶斯法

 1 // load data
 2 ArffLoader loader = new ArffLoader();
 3 loader.setFile(new File(args[0]));
 4 Instances structure = loader.getStructure();
 5 structure.setClassIndex(structure.numAttributes() - 1);
 6  
 7 // train NaiveBayes
 8 NaiveBayesUpdateable nb = new NaiveBayesUpdateable();
 9 NaiveBayes nb2 = new NaiveBayes();
10 nb.buildClassifier(structure);
11 Instance current;
12 while ((current = loader.getNextInstance(structure)) != null)
13     nb.updateClassifier(current);
14  
15 // output generated model
16 System.out.println(nb);