3

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

前もって感謝します、

4

2 に答える 2