学习weka（3）：idea中集成weka

最新推荐文章于 2026-04-25 12:58:55 发布

原创

最新推荐文章于 2026-04-25 12:58:55 发布 · 5.1k 阅读

标签

#java集成weka #java机器学习模块 #weka #idea

本文详细介绍如何使用Java调用WEKA进行数据预处理、模型训练及评估等操作，包括依赖引入、数据加载、模型训练与测试等多种实用技巧。

一、前期准备

1、加入依赖

百度搜索 maven，找到 maven 的 repository 仓库，寻找 weka 的依赖包：

在这里插入图片描述

找到稳定版本（我下载 weka 软件的时候就是这个版本，没有犹豫，就是它了），点击进去选择对应版本，依然是和自己软件版本一致：

在这里插入图片描述

点击进去可以看到对应的依赖语句代码了，当然也可以下载 jar 包，不过能偷懒就偷懒一下吧：

在这里插入图片描述

<dependency>
    <groupId>nz.ac.waikato.cms.weka</groupId>
    <artifactId>weka-stable</artifactId>
    <version>3.8.5</version>
</dependency>

2、新建 spring 项目，加入 weka 依赖

这里我创建的是 SpringBoot 项目，具体过程就不说了，超级简单。

创建一个 test 包，在建一个相关类就可以开始搞起了。

3、初步尝试

在网上找了半天，发现 weka 的实例代码比较少，搜关键词一大半网页都是软件 weka 的介绍，不得不说 java 搞 weka 实在是太过于小众了。

最终，皇天不负有心人，还是找到了一个代码实例：

public class test {
   
   
    public static void main(String[] args) {
   
   
        Instances ins = null;
        Classifier cfs = null;
        try {
   
   
            // read the training set
            File file = new File("D:\\weka\\Weka-3-8-5\\data\\weather.numeric.arff");
            ArffLoader loader = new ArffLoader();   //ArffLoader类是weka.core.converters下的
            loader.setFile(file);
            ins = loader.getDataSet();
            System.out.println(ins.numAttributes());
            ins.setClassIndex(ins.numAttributes() - 1);

            // 初始化分类器
            cfs = (Classifier) Class.forName("weka.classifiers.bayes.NaiveBayes").newInstance();

            // 使用训练集对数据集训练
            cfs.buildClassifier(ins);

            // 使用测试数据集测试分类器的性能
            Instance testInst;

            Evaluation testingEvaluation = new Evaluation(ins);
            int length = ins.numInstances(); //得到数据集样本个数

            for (int i = 0; i < length; i++) {
   
   
                testInst = ins.instance(i);
                testingEvaluation.evaluateModelOnceAndRecordPrediction(cfs, testInst);
            }

            // print the classifying results
            System.out.println("分类正确率:" + (1 - testingEvaluation.errorRate()));
        } catch (Exception e) {
   
   
            e.printStackTrace();
        }
    }
}

有了代码实例，接下来就相对容易一些，至少有一个大体框架的实例可供参考。

二、逐步提升

1、任务：加载 arff 文件、csv 与 arff 文件之间转换

加载 arff 的两种方法：

//方法一：使用DataSource类的read方法来加载arff文件
Instances data1=DataSource.read("data/weather.nominal.arff");

//方法二：使用直接制定加载器的方法来加载Arff文件
ArffLoader arffLoader=new ArffLoader();//创建ArffLoader实例
arffLoader.setSource(new File("data/weather.nominal.arff"));
Instances data2=arffLoader.getDataSet(

最低0.47元/天解锁文章