朴素贝叶斯算法Java实现

最新推荐文章于 2024-08-22 09:36:28 发布

amy6chunxiao

最新推荐文章于 2024-08-22 09:36:28 发布

阅读量1.3k

点赞数 1

分类专栏： Algorithm 文章标签： java 算法贝叶斯

本文链接：https://blog.csdn.net/u012673501/article/details/41483695

版权

本文介绍了如何使用Java编程实现朴素贝叶斯算法，包括构造分类器、判断测试样本类别、录入样本以及主方法的详细步骤。通过实例解析，帮助读者理解朴素贝叶斯算法的应用。

摘要由CSDN通过智能技术生成

一、构造分类器

/**
	 * 训练集分类（以每一行最后一个元素作为键，相同的键对应的行存入到同一链表中）
	 * 
	 * @param input
	 *            训练元祖
	 * @return 分类器
	 */
	public HashMap<Double, ArrayList<ArrayList<Double>>> practiceClassifyer(
			ArrayList<ArrayList<Double>> input) {

		HashMap<Double, ArrayList<ArrayList<Double>>> classifyer = new HashMap<Double, ArrayList<ArrayList<Double>>>();
		for (ArrayList<Double> arrayList : input) {

			if (arrayList.isEmpty())
				break;

			// 以最后一个数字作为该类别的键
			double key = arrayList.get(arrayList.size() - 1);
			if (classifyer.containsKey(key)) {

				ArrayList<ArrayList<Double>> temp = classifyer.get(key);
				temp.add(arrayList);
			} else {

				ArrayList<ArrayList<Double>> temp = new ArrayList<ArrayList<Double>>();
				temp.add(arrayList);
				classifyer.put(key, temp);
			}
		}
		return