对于朴素贝叶斯算法相信做数据挖掘和推荐系统的小伙们都耳熟能详了,算法原理我就不啰嗦了。我主要想通过java代码实现朴素贝叶斯算法,思想:
1. 用javabean +Arraylist 对于训练数据存储
2. 对于样本数据训练
具体的代码如下:
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