Facebook
From Tacky Gibbon, 4 Years ago, written in Plain Text.
Embed
Download Paste or View Raw
Hits: 216
  1. import java.io.BufferedReader;
  2. import java.io.FileNotFoundException;
  3. import java.io.FileReader;
  4. import java.util.Random;
  5.  
  6. import weka.classifiers.Evaluation;
  7. import weka.classifiers.bayes.NaiveBayes;
  8. import weka.core.Instances;
  9.  
  10. public class WekaApi {
  11.  
  12.         public static void main(String[] args) throws Exception {
  13.                 // TODO Auto-generated method stub
  14.                 BufferedReader breader = null;
  15.                
  16.                 breader = new BufferedReader(new FileReader("c:/pogoda.arff"));
  17.                
  18.                 Instances train = new Instances (breader);
  19.                 train.setClassIndex(train.numAttributes()-1);
  20.                
  21.                 breader.close();
  22.                
  23.                 NaiveBayes nB = new NaiveBayes();
  24.                 nB.buildClassifier(train);
  25.                 Evaluation eval = new Evaluation(train);
  26.                 eval.crossValidateModel(nB, train, 10, new Random(1));
  27.                 System.out.println(eval.toSummaryString("\nResults\n======\n",true));
  28.                 System.out.println(eval.fMeasure(1) + "" + eval.precision(1)+ "" + eval.recall(1));
  29.                
  30.                
  31.  
  32.         }
  33.  
  34. }
  35.