簡體   English   中英

R中e1071包的svm函數中概率= TRUE的函數

[英]Functionality of probability=TRUE in svm function of e1071 package in R

在R中, e1071包的svm函數中probability=TRUE的功能是什么?

model <- svm (Type ~ ., data, probability=TRUE, cost = 100, gamma = 1)

對於每個預測,將模型擬合和預測返回的probability參數設置為TRUE ,屬於響應變量的每個類的概率向量。 它們存儲在矩陣中,作為預測對象的屬性。

例如:

library(e1071)

model <- svm(Species ~ ., data = iris, probability=TRUE)
# (below I'm just predicting to the training dataset - it could of course just 
# as easily be a separate test dataset)
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

但請注意,為svm調用設置probability=TRUE很重要,而不僅僅是調用predict ,因為后者單獨會產生:

#      setosa versicolor virginica
# 1 0.3333333  0.3333333 0.3333333
# 2 0.3333333  0.3333333 0.3333333
# 3 0.3333333  0.3333333 0.3333333
# 4 0.3333333  0.3333333 0.3333333
# 5 0.3333333  0.3333333 0.3333333
# 6 0.3333333  0.3333333 0.3333333

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM