首先,对我的英语感到抱歉,我是巴西人,并且我正在改善它。
我正在使用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)
J48中的错误(分类〜。,数据= treino1,概率= TRUE):未使用的参数(概率= TRUE)
谢谢!
最佳答案
您应该指定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