Как выяснить, какие столбцы влияют на прогноз с помощью модели KNN Карета - PullRequest
Новая
       14

Как выяснить, какие столбцы влияют на прогноз с помощью модели KNN Карета

0 голосов
/ 28 мая 2019

У меня есть набор данных, в котором мне нужно предсказать столбец и выяснить, какие столбцы имеют отношение к прогнозу.

Мой код и модель выглядит так: 

set.seed(1)
trn_index = createDataPartition(y = low_birth_weight$id, p = 0.70, list = FALSE)
trn_low_bw = low_birth_weight[trn_index, ]
tst_low_bw = low_birth_weight[-trn_index, ]
low_bw_knn = train(low ~ ., method = "knn", data = trn_low_bw,
                   trControl = trainControl(method = 'cv', number = 5))

Модель достигает хорошей точности, но мне еще предстоит выяснить, как я могу увидеть соответствие каждого столбца прогнозу. Кто-нибудь знает где посмотреть?

Спасибо

...