我正在使用插入符号包来运行gbm模型。运行模型后,我将使用varImp函数提取重要预测变量的列表(显示前20个)。但是,我想在字符列表中捕获预测变量的名称。我怎么做?从varImp返回的对象似乎没有列出预测变量名称的属性-仅变量重要性。这是一个示例:
gbmModel= train(target ~. , data = trainData, ....other params )
varimp = varImp(gbmModel, scale=TRUE)
str(varimp)
List of 3
$ importance:'data.frame': 77 obs. of 1 variable:
..$ Overall: num [1:77] 6.63 0 5.35 2.01 0 ...
$ model : chr "gbm"
$ calledFrom: chr "varImp"
- attr(*, "class")= chr "varImp.train"
## Display the important variables
varimp
gbm variable importance
only 20 most important variables shown (out of 77)
Overall
Var126 100.000
Var189 99.647
Var113 41.994
... And so on
我想要一个清单(“ Var126”,“ Var189”,“ Var113” ...)
提前致谢,
最佳答案
从您提供的输出结构来看,
rownames(varimp$importance)
将返回您要的值。