2015-11-05 129 views
0

我想运行naieve bayes分类器。我已经在韦卡训练了这个模型。现在对于每个测试实例,我希望找到预测类别和概率分布,以便使用此值我也可以对实例进行排名。 我正在运行java的weka。我怎样才能得到weka java的概率分布?weka中naieve bayes的概率分布

回答

2
double[] predictionDistribution = null; 
    try { 
     double clsLabel = classifier.classifyInstance(Test Instance Here); 
     predictionDistribution = classifier.distributionForInstance(Test Instance Here); 
    } catch (Exception e) { 
     System.out.println("Unable to classify item\n"); 
    } 

使用double[]数组,然后用你的分类的distributionForInstance()方法。