如何在支持向量机中找到重要因素

时间:2016-01-14 03:57:14

标签: r machine-learning svm

原始数据很大,所以我不能在这里发布。 问题是我使用R中的包e1071来进行支持向量机分析。 原始数据有100个因子,预测结果为1或0。 例如,我生成一个包含10个因子的随机数据帧。

for (i in 1:10){
    factor<-c(factor,runif(10,5,10))
}
value<-matrix(factor,nrow=10)
y<-sample(0:1,10,replace=T)
data<-as.data.frame(cbind(y,value))

我做了预测工作,但我想知道如何确定哪些因素(10个因素中)对结果很重要(更相关)。

例如,结果可能是因子2,4,5,而10则对最终结果有贡献。

你能帮我解决这个问题吗? 非常感谢你。

1 个答案:

答案 0 :(得分:3)

这个问题的完整答案并不简单。以下是开始此主题的示例:

library(rpart)
library(e1071)

cat('Regression tree case:\n')
fit1 <- rpart(Species ~ ., data=iris)
print(fit1$variable.importance)

cat('SVM model case:\n')
fit2 <- svm(Species ~ ., data = iris)
w <- t(fit2$coefs) %*% fit2$SV                 # weight vectors
w <- apply(w, 2, function(v){sqrt(sum(v^2))})  # weight
w <- sort(w, decreasing = T)
print(w)

以上脚本的结果是:

Regression tree case:
 Petal.Width Petal.Length Sepal.Length  Sepal.Width 
    88.96940     81.34496     54.09606     36.01309 

SVM model case:
Petal.Length  Petal.Width Sepal.Length  Sepal.Width 
   12.160093    11.737364     6.623965     4.722632 

您可以看到两个模型的结果变量重要性相似。

这是解释SVM结果的众多方法之一。

有关详细信息,请参阅以下文章:“变量和特征选择简介”,http://jmlr.csail.mit.edu/papers/v3/guyon03a.html