Как да сортирате различна колона от функцията varImp в R

Използвам функцията varImp(model, scale = FALSE) за мултиклас и получих резултат от 20 най-важни променливи с три колони, тъй като има три класа. В момента е сортиран въз основа на първата колона, но има ли начин да го сортирате въз основа на втората колона? Опитах да го сортирам и конвертирам в dataframe, не работи: не може да принуди класа „varImp.train“ към data.frame.


r r-caret random-forest
person musik lover    schedule 06.11.2020    source източник
comment
Можете ли да предоставите възпроизводим пример и да покажете очаквания резултат за него?   -  person Ronak Shah    schedule 06.11.2020

Отговори (1)


arrow_upward
1
arrow_downward

Предполагам, че използвате randomForest с каретка, в противен случай функцията varImp() няма да работи.

моля, предоставете тази информация във вашия въпрос.

Така че можете да съхранявате резултатите от varImp() и да сортирате след:

library(caret)
rf = train(Species ~ .,data=iris,method="rf",trControl=trainControl(method="cv"),importance=TRUE)

varImp(rf,scale=FALSE)
rf variable importance

  variables are sorted by maximum importance across the classes
             setosa versicolor virginica
Petal.Length 22.013    33.4341    27.738
Petal.Width  22.454    32.3445    30.897
Sepal.Length  6.649     8.6931     8.569
Sepal.Width   4.420    -0.8063     3.560

res = varImp(rf,scale=FALSE)$importance

Сортиране по втора колона,

res[order(res$versicolor,decreasing=TRUE),]

                setosa versicolor virginica
Petal.Length 22.013431  33.434069 27.738319
Petal.Width  22.453571  32.344497 30.897214
Sepal.Length  6.649072   8.693132  8.569421
Sepal.Width   4.419712  -0.806256  3.559570
person StupidWolf    schedule 06.11.2020
comment
Благодаря ти много! - person musik lover; 07.11.2020