在Weka分类器

时间:2016-12-10 13:12:53

标签: machine-learning weka

我使用以下代码来使用weka库实现活动识别。我使用随机森林分类器,这是一个多类分类问题。我使用训练数据集来训练随机森林模型。之后,我使用该模型对不包含类标签的测试数据集进行分类。作为班级,我得到双倍的价值。如何获得正确的整数值?有人可以帮助我吗? 而且如果我将得到的双值转换为int值,我只得到相同的整数值(0),因为预测的双精度值介于0和1之间。

    BufferedReader br = null;

            br = new BufferedReader(new FileReader("/home/thamali/Desktop/WekaProject/output.arff"));

            Instances trainData = new Instances(br);
            trainData.setClassIndex(trainData.numAttributes() - 1);
            br.close();
            RandomForest rf = new RandomForest();
            rf.setNumTrees(23);
            rf.setMaxDepth(18);
            rf.buildClassifier(trainData);

            BufferedReader br1 = new BufferedReader(new FileReader("/home/thamali/Desktop/WekaProject/testData.arff"));
            Instances testData = new Instances(br1);
            testData.setClassIndex(testData.numAttributes() - 1);

            double value=rf.classifyInstance(testData.instance(0));
            System.out.print(value);

0 个答案:

没有答案