预测函数将概率在0到1之间转换并存储在数据框中

时间:2018-11-28 11:51:47

标签: r

Predict函数代码返回这样的输出。

library(e1071)
model <- svm(Species ~ ., data = iris, probability=TRUE)
pred <- predict(model, iris, probability=TRUE)

head(attr(pred, "probabilities"))
#      setosa versicolor   virginica
# 1 0.9803339 0.01129740 0.008368729
# 2 0.9729193 0.01807053 0.009010195
# 3 0.9790435 0.01192820 0.009028276
# 4 0.9750030 0.01531171 0.009685342
# 5 0.9795183 0.01164689 0.008834838
# 6 0.9740730 0.01679643 0.009130620

所以,我写了一段这样的代码:-

Code:-

pred_df <- as.data.frame(pred)

这将返回这样的输出,(刚刚计算出值)

1 Setosa
2 Versicolor
3 Virginica
4 Setosa
5 Setosa

但是我的首选输出将是这样(刚刚组成了值),

   Setosa      Versicolor       Virginica
1 0.62          0.11               0.27
2 0.41          0.55               0.04

***Pred is a factor and Pred_df is a dataframe***

我希望以小数形式而不是整数形式返回数字。请帮助我。

1 个答案:

答案 0 :(得分:1)

要查看发生了什么,您需要查看结构内部。

str(pred)
 Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 - attr(*, "names")= chr [1:150] "1" "2" "3" "4" ...
 - attr(*, "probabilities")= num [1:150, 1:3] 0.979 0.971 0.978 0.973 0.978 ...
  ..- attr(*, "dimnames")=List of 2
  .. ..$ : chr [1:150] "1" "2" "3" "4" ...
  .. ..$ : chr [1:3] "setosa" "versicolor" "virginica"

所以

as.data.frame(attr(pred,"probabilities"))

应该做你想要的。