概率分类 - 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

概率分类 - R

Probabilities Classification - R

r

machine-learning

svm

weka

j48