代码-Weka的决策树类J48

最新推荐文章于 2021-02-28 01:32:24 发布

转载最新推荐文章于 2021-02-28 01:32:24 发布 · 358 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/lam99v/p/3503198.html

文章标签：

#数据结构与算法 #java

本文介绍了一个使用Weka库中的J48决策树分类器进行分类任务的Java程序示例。该示例加载了CPU数据集，并通过训练模型来评估J48决策树的分类精度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package kit.weka;

/**

* desc:试试Weka的决策树类

* <code>J48Test</code>

*/

import java.io.File;

import java.io.IOException;

import weka.classifiers.Classifier;

import weka.classifiers.trees.J48;

import weka.core.Instances;

import weka.core.converters.ArffLoader;

public class J48Test {

/**

     * @param args

     * @throws Exception

*/

    public static void main(String[] args) throws Exception {

         Classifier m_classifier = new J48();

            File inputFile = new File("d:\\Program Files\\Weka-3-7\\data\\cpu.with.vendor.arff");//训练语料文件

            ArffLoader atf = new ArffLoader();

            atf.setFile(inputFile);

            Instances instancesTrain = atf.getDataSet(); // 读入训练文件

            inputFile = new File("d:\\Program Files\\Weka-3-7\\data\\cpu.with.vendor.arff");//测试语料文件

            atf.setFile(inputFile);

            Instances instancesTest = atf.getDataSet(); // 读入测试文件

            instancesTest.setClassIndex(0); //设置分类属性所在行号（第一行为0号），instancesTest.numAttributes()可以取得属性总数

            double sum = instancesTest.numInstances(),//测试语料实例数

            right = 0.0f;

            instancesTrain.setClassIndex(0);

             m_classifier.buildClassifier(instancesTrain); //训练

            for(int  i = 0;i<sum;i++)//测试分类结果

                if(m_classifier.classifyInstance(instancesTest.instance(i))==instancesTest.instance(i).classValue())//如果预测值和答案值相等（测试语料中的分类列提供的须为正确答案，结果才有意义）

                  right++;//正确值加1

            System.out.println("J48 classification precision:"+(right/sum));

转载于:https://www.cnblogs.com/lam99v/p/3503198.html