Java实现的朴素贝叶斯算法示例

package NB;
/**
 * 训练样本的属性 javaBean
 *
 */
public class JavaBean {
 int age;
 String income;
 String student;
 String credit_rating;
 String buys_computer;
 public JavaBean(){
 }
public JavaBean(int age,String income,String student,String credit_rating,String buys_computer){
  this.age=age;
  this.income=income;
  this.student=student;
  this.credit_rating=credit_rating;
  this.buys_computer=buys_computer;
}
public int getAge() {
  return age;
}
public void setAge(int age) {
  this.age = age;
}
public String getIncome() {
  return income;
}
public void setIncome(String income) {
  this.income = income;
}
public String getStudent() {
  return student;
}
public void setStudent(String student) {
  this.student = student;
}
public String getCredit_rating() {
  return credit_rating;
}
public void setCredit_rating(String credit_rating) {
  this.credit_rating = credit_rating;
}
public String getBuys_computer() {
  return buys_computer;
}
public void setBuys_computer(String buys_computer) {
  this.buys_computer = buys_computer;
}
@Override
public String toString() {
  return "JavaBean [age=" + age + ", income=" + income + ", student="
      + student + ", credit_rating=" + credit_rating + ", buys_computer="
      + buys_computer + "]";
}
}

package NB;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.util.ArrayList;
public class TestNB {
  /**data_length
   * 算法的思想
   */
  public static ArrayList list = new ArrayList();;
  static int data_length=0;
  public static void main(String[] args) {
    // 1.读取数据，放入list容器中
    File file = new File("E://test.txt");
    txt2String(file);
    //数据测试样本
    testData(25,"Medium","Yes","Fair");
  }
  // 读取样本数据
  public static void txt2String(File file) {
    try {
      BufferedReader br = new BufferedReader(new FileReader(file));// 构造一个BufferedReader类来读取文件
      String s = null;
      while ((s = br.readLine()) != null) {// 使用readLine方法，一次读一行
        data_length++;
        splitt(s);
      }
      br.close();
    } catch (Exception e) {
      e.printStackTrace();
    }
  }
  // 存入ArrayList中
   public static void splitt(String str){
      String strr = str.trim();
      String[] abc = strr.split("[\\p{Space}]+");
      int age=Integer.parseInt(abc[0]);
      JavaBean bean=new JavaBean(age, abc[1], abc[2], abc[3], abc[4]);
      list.add(bean);
    }
   // 训练样本，测试
   public static void testData(int age,String a,String b,String c){
     //训练样本
     int number_yes=0;
     int bumber_no=0;
     // age情况 个数
     int num_age_yes=0;
     int num_age_no=0;
     // income
     int num_income_yes=0;
     int num_income_no=0;
     // student
     int num_student_yes=0;
     int num_stdent_no=0;
     //credit
     int num_credit_yes=0;
     int num_credit_no=0;
     //遍历List 获得数据
     for(int i=0;inb_buy_no){
        System.out.println("新用户买的概率大");
      }else {
        System.out.println("新用户不买的概率大");
      }
   }
}