Проблемы создания сюжета для varImp

Question

library(tidyverse)
library(ggplot2)
library(caret)
varImp(fullFit, scale = FALSE)
plot(varImp(fullFit, scale = FALSE))

> varImp(fullFit, scale = FALSE)
                 Overall
bedrooms       2.9003032
bathrooms      1.9974678
sqft_living   13.5250547
sqft_lot       0.7873693
floors         2.2513678
waterfront    11.2258728
sqft_above     3.4476093
sqft_living15  6.4453136
sqft_lot15     1.9466399

Когда я пытаюсь построить VarImp, он не отображает переменные (спальни, ванные комнаты и т. Д.). Как правильно создать визуализацию, отображающую значение переменной из пакета каретки?

plot(varImp(fullFit, scale = FALSE))

Answer 1

set.seed(23)
library(caret)
library(mlbench)
data(Sonar)
traincontrol <- trainControl(method = "cv", number = 5, returnResamp = "all",   savePredictions='all', classProbs = TRUE, summaryFunction = twoClassSummary)

nnet_model <- train(Class ~.,
                 data = Sonar,
                 method = "nnet", 
                 trControl = traincontrol,
                 metric = "ROC", 
                 trace = FALSE)

plot(varImp(nnet_model$finalModel, scale = FALSE))#wrong image
plot(varImp(nnet_model, scale = FALSE))#correct image