Java实现朴素贝叶斯分类器

来源：互联网发布：火烈鸟化妆品知乎编辑：程序博客网时间：2024/05/17 03:03

实验描述:

对指定数据集进行分类问题的分析，选择适当的分类算法，编写程序实现，提交程序和结果报告

数据集： balance-scale.data（见附件一），已有数据集构建贝叶斯分类器。

数据包括四个属性：五个属性值第一个属性值表示样本的类别号，其他四个属性为四个不同的特征值。

实验环境和编程语言：

本实验使用的编程语言为：Java

编程环境为：Intellij idea

构建分类器的算法为：朴素贝叶斯算法

训练的样本个数为：625个

样本示例：R,1,1,1,2（表示属于R类，四各属性值为1 1 1 2）

训练样本和测试样本见附件一。

工程和测试数据下载地址：

http://download.csdn.net/detail/qq_24369113/9711643

实验结果分析：

根据程序运行的结果如图1-1所示，使用训练的样本对分类器进行检测，分类得到295个R类（实际288个），330个L类（实际288个），0个B类（实际49个），综合分类正确率为92.16%。可以认为分类器有较高的正确率，但是对于类B的分类还是有一定的误差，具体原因会在思考与改进中说明。

思考与改进：

思考点：

本次实验编程利用朴素贝叶斯算法构建的分类器，虽然整体情况是拥有较高的分类正确率（92.16%），但是对训练样本再检测时没有发现一个B类样本，出现的较大的偏差，分析原因如下：

1. B类样本点总数很少，以至于P（y_B）就远远小于P（y_R）和P（y_L），所以对于后验概率公式，P（y_B|x）必然比P（y_L|x）和P（y_R|x）要小，因此导致最后样本难以分到B类。

2. 观察训练样本集可以发现，B类的样本点和L类以及R类并没有很大区别，其不具备很强的统计可分性，所以对于依赖于统计概率的贝叶斯方法无法体现出其优势，所以导致了最终没有样本被分到B类中。

改进点：

本算法在实现贝叶斯分类的时候，每一次都进行了遍历样本集来求出

P（a₁|y₁）、P（a₂|y₁）……P（a_n|y_n），所以导致算法的整体时间复杂度较高，在面临较大的数据集时可能就会出现问题，因此可以考虑实现将已经计算过的

P（a_x|y_x）存储到变量或者数组中，在下一次计算的时候可以直接调用，从而可以避免再一次的遍历训练集来计算概率，实现时间复杂度的降低。

编程实现：

package com.company;/******************************************************************************** * *  * 数据挖掘实验，分类实验 * 朴素贝叶斯分类 * *训练样本示例格式：（R,1,5,5,4） */import java.io.File;import java.util.Scanner;import java.util.Vector;public class Bayes {    static Vector<String> indata = new Vector<>();//读入数据    static Vector<int[]> catagory_R = new Vector<>();//存储类别R的所有数据    static Vector<int[]> catagory_L = new Vector<>();//存储类别L的所有数据    static Vector<int[]> catagory_B = new Vector<>();//存储类别B的所有数据    public static boolean loadData(String url) {//加载测试的数据文件        try {            Scanner in = new Scanner(new File(url));//读入文件            while (in.hasNextLine()) {                String str = in.nextLine();//将文件的每一行存到str的临时变量中                indata.add(str);//将每一个样本点的数据追加到Vector 中            }            return true;        } catch (Exception e) { //如果出错返回false            return false;        }    }    public static void pretreatment(Vector<String> indata) {   //数据预处理，将原始数据中的每一个属性值提取出来存放到Vector<double[]>  data中        int i = 0;        String t;        while (i < indata.size()) {//取出indata中的每一行值            int[] tem = new int[4];            t = indata.get(i);            String[] sourceStrArray = t.split(",", 5);//使用字符串分割函数提取出各属性值            switch (sourceStrArray[0]) {                case "R": {                    for (int j = 1; j < 5; j++) {                        tem[j - 1] = Integer.parseInt(sourceStrArray[j]);                    }                    catagory_R.add(tem);                    break;                }                case "L": {                    for (int j = 1; j < 5; j++) {                        tem[j - 1] = Integer.parseInt(sourceStrArray[j]);                    }                    catagory_L.add(tem);                    break;                }                case "B": {                    for (int j = 1; j < 5; j++) {                        tem[j - 1] = Integer.parseInt(sourceStrArray[j]);                    }                    catagory_B.add(tem);                    break;                }            }            i++;        }    }    public static double bayes(int[] x, Vector<int[]> catagory) {        double[] ai_y = new double[4];        int[] sum_ai = new int[4];        for (int i = 0; i < 4; i++) {            for (int j = 0; j < catagory.size(); j++) {                if (x[i] == catagory.get(j)[i])                    sum_ai[i]++;            }        }        for (int i = 0; i < 4; i++) {            ai_y[i] = (double) sum_ai[i] / (double) catagory.size();        }        return ai_y[0] * ai_y[1] * ai_y[2] * ai_y[3];    }    public static void main(String[] args) {        loadData("balance-scale.data");        pretreatment(indata);        double p_yR = (double) catagory_R.size() / (double) (indata.size());//表示概率p（R）        double p_yB = (double) catagory_B.size() / (double) (indata.size());//表示概率p（B）        double p_yL = (double) catagory_L.size() / (double) (indata.size());//表示概率p（L）        int[] x = new int[4];        double x_in_R, x_in_L, x_in_B;        int sumR=0, sumL=0, sumB=0;        double correct=0;        System.out.println("请输入样本x格式如下：\n 1 1 1 1\n");        int r = 0;        while (r < indata.size()) {            for (int i = 0; i < 4; i++)                //读取数字放入数组的第i个元素                x[i] = Integer.parseInt(indata.get(r).split(",", 5)[i + 1]);            x_in_B = bayes(x, catagory_B) * p_yB;            x_in_L = bayes(x, catagory_L) * p_yL;            x_in_R = bayes(x, catagory_R) * p_yR;            if (x_in_B == Math.max(Math.max(x_in_B, x_in_L), x_in_R)) {                System.out.println("输入的第"+r+"样本属于类别：B");                sumB++;                if(indata.get(r).split(",",5)[0].equals("B"))                    correct++;            } else if (x_in_L == Math.max(Math.max(x_in_B, x_in_L), x_in_R)) {                System.out.println("输入的第"+r+"样本属于类别：L");                sumL++;                if(indata.get(r).split(",",5)[0].equals("L"))                    correct++;            } else if (x_in_R == Math.max(Math.max(x_in_B, x_in_L), x_in_R)) {                System.out.println("输入的第"+r+"样本属于类别：R");                sumR++;                if(indata.get(r).split(",",5)[0].equals("R"))                    correct++;            }            r++;        }        System.out.println("使用训练样本进行分类器检验得到结果统计如下：");        System.out.println("R类有："+sumR+"    实际有R类样本"+catagory_R.size()+"个");        System.out.println("L类有："+sumL+"    实际有L类样本"+catagory_L.size()+"个");        System.out.println("B类有："+sumB+"      实际有B类样本"+catagory_B.size()+"个");        System.out.println("分类的正确率为"+correct*1.0/indata.size()*100+"%");    }}

（附件一）训练样本和测试样本集：

B,1,1,1,1R,1,1,1,2R,1,1,1,3R,1,1,1,4R,1,1,1,5R,1,1,2,1R,1,1,2,2R,1,1,2,3R,1,1,2,4R,1,1,2,5R,1,1,3,1R,1,1,3,2R,1,1,3,3R,1,1,3,4R,1,1,3,5R,1,1,4,1R,1,1,4,2R,1,1,4,3R,1,1,4,4R,1,1,4,5R,1,1,5,1R,1,1,5,2R,1,1,5,3R,1,1,5,4R,1,1,5,5L,1,2,1,1B,1,2,1,2R,1,2,1,3R,1,2,1,4R,1,2,1,5B,1,2,2,1R,1,2,2,2R,1,2,2,3R,1,2,2,4R,1,2,2,5R,1,2,3,1R,1,2,3,2R,1,2,3,3R,1,2,3,4R,1,2,3,5R,1,2,4,1R,1,2,4,2R,1,2,4,3R,1,2,4,4R,1,2,4,5R,1,2,5,1R,1,2,5,2R,1,2,5,3R,1,2,5,4R,1,2,5,5L,1,3,1,1L,1,3,1,2B,1,3,1,3R,1,3,1,4R,1,3,1,5L,1,3,2,1R,1,3,2,2R,1,3,2,3R,1,3,2,4R,1,3,2,5B,1,3,3,1R,1,3,3,2R,1,3,3,3R,1,3,3,4R,1,3,3,5R,1,3,4,1R,1,3,4,2R,1,3,4,3R,1,3,4,4R,1,3,4,5R,1,3,5,1R,1,3,5,2R,1,3,5,3R,1,3,5,4R,1,3,5,5L,1,4,1,1L,1,4,1,2L,1,4,1,3B,1,4,1,4R,1,4,1,5L,1,4,2,1B,1,4,2,2R,1,4,2,3R,1,4,2,4R,1,4,2,5L,1,4,3,1R,1,4,3,2R,1,4,3,3R,1,4,3,4R,1,4,3,5B,1,4,4,1R,1,4,4,2R,1,4,4,3R,1,4,4,4R,1,4,4,5R,1,4,5,1R,1,4,5,2R,1,4,5,3R,1,4,5,4R,1,4,5,5L,1,5,1,1L,1,5,1,2L,1,5,1,3L,1,5,1,4B,1,5,1,5L,1,5,2,1L,1,5,2,2R,1,5,2,3R,1,5,2,4R,1,5,2,5L,1,5,3,1R,1,5,3,2R,1,5,3,3R,1,5,3,4R,1,5,3,5L,1,5,4,1R,1,5,4,2R,1,5,4,3R,1,5,4,4R,1,5,4,5B,1,5,5,1R,1,5,5,2R,1,5,5,3R,1,5,5,4R,1,5,5,5L,2,1,1,1B,2,1,1,2R,2,1,1,3R,2,1,1,4R,2,1,1,5B,2,1,2,1R,2,1,2,2R,2,1,2,3R,2,1,2,4R,2,1,2,5R,2,1,3,1R,2,1,3,2R,2,1,3,3R,2,1,3,4R,2,1,3,5R,2,1,4,1R,2,1,4,2R,2,1,4,3R,2,1,4,4R,2,1,4,5R,2,1,5,1R,2,1,5,2R,2,1,5,3R,2,1,5,4R,2,1,5,5L,2,2,1,1L,2,2,1,2L,2,2,1,3B,2,2,1,4R,2,2,1,5L,2,2,2,1B,2,2,2,2R,2,2,2,3R,2,2,2,4R,2,2,2,5L,2,2,3,1R,2,2,3,2R,2,2,3,3R,2,2,3,4R,2,2,3,5B,2,2,4,1R,2,2,4,2R,2,2,4,3R,2,2,4,4R,2,2,4,5R,2,2,5,1R,2,2,5,2R,2,2,5,3R,2,2,5,4R,2,2,5,5L,2,3,1,1L,2,3,1,2L,2,3,1,3L,2,3,1,4L,2,3,1,5L,2,3,2,1L,2,3,2,2B,2,3,2,3R,2,3,2,4R,2,3,2,5L,2,3,3,1B,2,3,3,2R,2,3,3,3R,2,3,3,4R,2,3,3,5L,2,3,4,1R,2,3,4,2R,2,3,4,3R,2,3,4,4R,2,3,4,5L,2,3,5,1R,2,3,5,2R,2,3,5,3R,2,3,5,4R,2,3,5,5L,2,4,1,1L,2,4,1,2L,2,4,1,3L,2,4,1,4L,2,4,1,5L,2,4,2,1L,2,4,2,2L,2,4,2,3B,2,4,2,4R,2,4,2,5L,2,4,3,1L,2,4,3,2R,2,4,3,3R,2,4,3,4R,2,4,3,5L,2,4,4,1B,2,4,4,2R,2,4,4,3R,2,4,4,4R,2,4,4,5L,2,4,5,1R,2,4,5,2R,2,4,5,3R,2,4,5,4R,2,4,5,5L,2,5,1,1L,2,5,1,2L,2,5,1,3L,2,5,1,4L,2,5,1,5L,2,5,2,1L,2,5,2,2L,2,5,2,3L,2,5,2,4B,2,5,2,5L,2,5,3,1L,2,5,3,2L,2,5,3,3R,2,5,3,4R,2,5,3,5L,2,5,4,1L,2,5,4,2R,2,5,4,3R,2,5,4,4R,2,5,4,5L,2,5,5,1B,2,5,5,2R,2,5,5,3R,2,5,5,4R,2,5,5,5L,3,1,1,1L,3,1,1,2B,3,1,1,3R,3,1,1,4R,3,1,1,5L,3,1,2,1R,3,1,2,2R,3,1,2,3R,3,1,2,4R,3,1,2,5B,3,1,3,1R,3,1,3,2R,3,1,3,3R,3,1,3,4R,3,1,3,5R,3,1,4,1R,3,1,4,2R,3,1,4,3R,3,1,4,4R,3,1,4,5R,3,1,5,1R,3,1,5,2R,3,1,5,3R,3,1,5,4R,3,1,5,5L,3,2,1,1L,3,2,1,2L,3,2,1,3L,3,2,1,4L,3,2,1,5L,3,2,2,1L,3,2,2,2B,3,2,2,3R,3,2,2,4R,3,2,2,5L,3,2,3,1B,3,2,3,2R,3,2,3,3R,3,2,3,4R,3,2,3,5L,3,2,4,1R,3,2,4,2R,3,2,4,3R,3,2,4,4R,3,2,4,5L,3,2,5,1R,3,2,5,2R,3,2,5,3R,3,2,5,4R,3,2,5,5L,3,3,1,1L,3,3,1,2L,3,3,1,3L,3,3,1,4L,3,3,1,5L,3,3,2,1L,3,3,2,2L,3,3,2,3L,3,3,2,4R,3,3,2,5L,3,3,3,1L,3,3,3,2B,3,3,3,3R,3,3,3,4R,3,3,3,5L,3,3,4,1L,3,3,4,2R,3,3,4,3R,3,3,4,4R,3,3,4,5L,3,3,5,1R,3,3,5,2R,3,3,5,3R,3,3,5,4R,3,3,5,5L,3,4,1,1L,3,4,1,2L,3,4,1,3L,3,4,1,4L,3,4,1,5L,3,4,2,1L,3,4,2,2L,3,4,2,3L,3,4,2,4L,3,4,2,5L,3,4,3,1L,3,4,3,2L,3,4,3,3B,3,4,3,4R,3,4,3,5L,3,4,4,1L,3,4,4,2B,3,4,4,3R,3,4,4,4R,3,4,4,5L,3,4,5,1L,3,4,5,2R,3,4,5,3R,3,4,5,4R,3,4,5,5L,3,5,1,1L,3,5,1,2L,3,5,1,3L,3,5,1,4L,3,5,1,5L,3,5,2,1L,3,5,2,2L,3,5,2,3L,3,5,2,4L,3,5,2,5L,3,5,3,1L,3,5,3,2L,3,5,3,3L,3,5,3,4B,3,5,3,5L,3,5,4,1L,3,5,4,2L,3,5,4,3R,3,5,4,4R,3,5,4,5L,3,5,5,1L,3,5,5,2B,3,5,5,3R,3,5,5,4R,3,5,5,5L,4,1,1,1L,4,1,1,2L,4,1,1,3B,4,1,1,4R,4,1,1,5L,4,1,2,1B,4,1,2,2R,4,1,2,3R,4,1,2,4R,4,1,2,5L,4,1,3,1R,4,1,3,2R,4,1,3,3R,4,1,3,4R,4,1,3,5B,4,1,4,1R,4,1,4,2R,4,1,4,3R,4,1,4,4R,4,1,4,5R,4,1,5,1R,4,1,5,2R,4,1,5,3R,4,1,5,4R,4,1,5,5L,4,2,1,1L,4,2,1,2L,4,2,1,3L,4,2,1,4L,4,2,1,5L,4,2,2,1L,4,2,2,2L,4,2,2,3B,4,2,2,4R,4,2,2,5L,4,2,3,1L,4,2,3,2R,4,2,3,3R,4,2,3,4R,4,2,3,5L,4,2,4,1B,4,2,4,2R,4,2,4,3R,4,2,4,4R,4,2,4,5L,4,2,5,1R,4,2,5,2R,4,2,5,3R,4,2,5,4R,4,2,5,5L,4,3,1,1L,4,3,1,2L,4,3,1,3L,4,3,1,4L,4,3,1,5L,4,3,2,1L,4,3,2,2L,4,3,2,3L,4,3,2,4L,4,3,2,5L,4,3,3,1L,4,3,3,2L,4,3,3,3B,4,3,3,4R,4,3,3,5L,4,3,4,1L,4,3,4,2B,4,3,4,3R,4,3,4,4R,4,3,4,5L,4,3,5,1L,4,3,5,2R,4,3,5,3R,4,3,5,4R,4,3,5,5L,4,4,1,1L,4,4,1,2L,4,4,1,3L,4,4,1,4L,4,4,1,5L,4,4,2,1L,4,4,2,2L,4,4,2,3L,4,4,2,4L,4,4,2,5L,4,4,3,1L,4,4,3,2L,4,4,3,3L,4,4,3,4L,4,4,3,5L,4,4,4,1L,4,4,4,2L,4,4,4,3B,4,4,4,4R,4,4,4,5L,4,4,5,1L,4,4,5,2L,4,4,5,3R,4,4,5,4R,4,4,5,5L,4,5,1,1L,4,5,1,2L,4,5,1,3L,4,5,1,4L,4,5,1,5L,4,5,2,1L,4,5,2,2L,4,5,2,3L,4,5,2,4L,4,5,2,5L,4,5,3,1L,4,5,3,2L,4,5,3,3L,4,5,3,4L,4,5,3,5L,4,5,4,1L,4,5,4,2L,4,5,4,3L,4,5,4,4B,4,5,4,5L,4,5,5,1L,4,5,5,2L,4,5,5,3B,4,5,5,4R,4,5,5,5L,5,1,1,1L,5,1,1,2L,5,1,1,3L,5,1,1,4B,5,1,1,5L,5,1,2,1L,5,1,2,2R,5,1,2,3R,5,1,2,4R,5,1,2,5L,5,1,3,1R,5,1,3,2R,5,1,3,3R,5,1,3,4R,5,1,3,5L,5,1,4,1R,5,1,4,2R,5,1,4,3R,5,1,4,4R,5,1,4,5B,5,1,5,1R,5,1,5,2R,5,1,5,3R,5,1,5,4R,5,1,5,5L,5,2,1,1L,5,2,1,2L,5,2,1,3L,5,2,1,4L,5,2,1,5L,5,2,2,1L,5,2,2,2L,5,2,2,3L,5,2,2,4B,5,2,2,5L,5,2,3,1L,5,2,3,2L,5,2,3,3R,5,2,3,4R,5,2,3,5L,5,2,4,1L,5,2,4,2R,5,2,4,3R,5,2,4,4R,5,2,4,5L,5,2,5,1B,5,2,5,2R,5,2,5,3R,5,2,5,4R,5,2,5,5L,5,3,1,1L,5,3,1,2L,5,3,1,3L,5,3,1,4L,5,3,1,5L,5,3,2,1L,5,3,2,2L,5,3,2,3L,5,3,2,4L,5,3,2,5L,5,3,3,1L,5,3,3,2L,5,3,3,3L,5,3,3,4B,5,3,3,5L,5,3,4,1L,5,3,4,2L,5,3,4,3R,5,3,4,4R,5,3,4,5L,5,3,5,1L,5,3,5,2B,5,3,5,3R,5,3,5,4R,5,3,5,5L,5,4,1,1L,5,4,1,2L,5,4,1,3L,5,4,1,4L,5,4,1,5L,5,4,2,1L,5,4,2,2L,5,4,2,3L,5,4,2,4L,5,4,2,5L,5,4,3,1L,5,4,3,2L,5,4,3,3L,5,4,3,4L,5,4,3,5L,5,4,4,1L,5,4,4,2L,5,4,4,3L,5,4,4,4B,5,4,4,5L,5,4,5,1L,5,4,5,2L,5,4,5,3B,5,4,5,4R,5,4,5,5L,5,5,1,1L,5,5,1,2L,5,5,1,3L,5,5,1,4L,5,5,1,5L,5,5,2,1L,5,5,2,2L,5,5,2,3L,5,5,2,4L,5,5,2,5L,5,5,3,1L,5,5,3,2L,5,5,3,3L,5,5,3,4L,5,5,3,5L,5,5,4,1L,5,5,4,2L,5,5,4,3L,5,5,4,4L,5,5,4,5L,5,5,5,1L,5,5,5,2L,5,5,5,3L,5,5,5,4B,5,5,5,5

0 0