У меня есть набор данных, в котором мне нужно предсказать столбец и выяснить, какие столбцы имеют отношение к прогнозу.
Мой код и модель выглядит так:
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))
Модель достигает хорошей точности, но мне еще предстоит выяснить, как я могу увидеть соответствие каждого столбца прогнозу. Кто-нибудь знает где посмотреть?
Спасибо