r - 概率分类 - R

Question

首先，对我的英语感到抱歉，我是巴西人，而且我还在提高它。

我正在使用 RWeka 创建分类模型(NaiveBayes、JRip、J48、SVM)，我需要每个实例的概率。

对于 SVM(包“e1041”)我只是这样做

model<-svm(classification~.,data=treino1, probability=TRUE)
pred <- predict(model, teste1[,-ncol(teste1)], probability = TRUE)
prob<-attr(pred, "probabilities")

然后，输出类似于:

           0           1
1    9.126349e-01 0.087365132
2    9.085960e-01 0.091404044
3    9.414368e-01 0.058563209

我应该怎样做才能获得与 J48(包“RWeka”)相同类型的输出？

model<-J48(classification~.,data=treino1, probability=TRUE)

Error in J48(classification ~ ., data = treino1, probability = TRUE) : unused argument (probability = TRUE)

谢谢!

Answer 1

您应该指定type = 'probability。以下是使用 iris 数据集的示例:

require(RWeka)
fit = J48(Species~., data=iris)
pred = predict(fit, iris, type = 'probability')
tail(p)

       setosa versicolor virginica
145      0 0.02173913 0.9782609
146      0 0.02173913 0.9782609
147      0 0.02173913 0.9782609
148      0 0.02173913 0.9782609
149      0 0.02173913 0.9782609
150      0 0.02173913 0.9782609