キャレット パッケージを使用して 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" ...) のようなリストが欲しい
前もって感謝します、