Переменная важность в карете

Question

Я использую пакет Caret в R для обучения модели логистической регрессии для задачи двоичной классификации.Я был в состоянии получить результаты, точность и т. Д., Но я также хочу важность переменных (в порядке убывания важности).Я использовал функцию varImp ().Но согласно документации, важность зависит от класса:

"Для большинства моделей классификации каждый предиктор будет иметь отдельную переменную важность для каждого класса (исключение составляют деревья классификации, деревья в пакетах и ​​деревья с повышенными параметрами)."

Как я могу получить значение переменной для каждого класса?

Спасибо

Answer 1

Для первой части вы пробовали:

round(importance(myModel$finalModel), 2)

Для того, чтобы поместить это в порядке убывания:

imp <- round(importance(myModel$finalModel), 2)
dfimp <- data.frame(feature = rownames(imp), MeanDecreaseGini = as.numeric(imp))
dfimp[order(dfimp$MeanDecreaseGini, decreasing = T),]