我使用以下代码来使用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);