It_BeeCoder的博客

不忘初心,方得始终

Java中调用weka的决策树算法


推荐于2016-10-09 19:33:53

1
2
3
4



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
public static void Regular() throws Exception {
        File inputfile = new File("F:\\weka\\eucalyptus_Train.arff");
        ArffLoader loader = new ArffLoader();
        loader.setFile(inputfile);         
        Instances insTrain = loader.getDataSet();
        insTrain.setClassIndex(insTrain.numAttributes()-1);
         
        inputfile = new File("F:\\weka\\eucalyptus_Test.arff");
        loader.setFile(inputfile);
        Instances insTest = loader.getDataSet();
        insTest.setClassIndex(insTest.numAttributes()-1);
         
        double sum = insTest.numInstances();
        int right = 0;
        Classifier clas = new J48();
        //Classifier clas = new weka.classifiers.bayes.BayesNet();
        clas.buildClassifier(insTrain);
         
        for(int i = 0; i < sum; i++) {
            if(clas.classifyInstance(insTest.instance(i)) == insTest.instance(i).classValue()) {
                right++;
            }
            System.out.println(clas.classifyInstance(insTest.instance(i))+" : "+insTest.instance(i).classValue());
        }
        System.out.println("分类准确率:"+right/sum);
    }
阅读更多
个人分类: 41 学术科研
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭