bayesian.java

implement bayes form spam filter (java source code)

/* Daniel Shiffman               */
/* Bayesian Spam Filter Example  */
/* Programming from A to Z       */
/* Spring 2007                   */
/* http://www.shiffman.net       */
/* [e-mail]       */

// Main driver program

//package bayes;

import java.io.*;
//import a2z.*;

public class Bayesian {
  public static void main (String args[]) {
    try {

      // Create a new SpamFilter Object
      SpamFilter filter = new SpamFilter();

      // Train spam with a file of spam e-mails
      filter.trainSpam("bal2an.txt");
      // Train spam with a file of regular e-mails
      filter.trainGood("tekno.txt");
      // We are finished adding words so finalize the results
      filter.finalizeTraining();


      for (int i = 1; i < 3; i++) {
        // Read in a text file
        A2ZFileReader fr = new A2ZFileReader("messages/pesan" + i + ".txt");
        String stuff = fr.getContent();

        // Ask the filter to analyze it
        boolean spam = filter.analyze(stuff);

        // Print results
        if (spam) System.out.println("iki bal2 an jeh");
        else System.out.println("lek iki tekno su");
      }

    } catch (IOException e) {
      e.printStackTrace();
    }
  }
}