функция поезда в пакете R продолжает сбой

Question

Я тренирую модель kNN с функцией поезда из пакета R Caret и набора данных из 2000 записей. Я использовал следующий код:

set.seed(400)
ctrl <- trainControl(method="none")
knnFit <- train(Class ~ ., data = ScaniaTrain, method = "knn", trControl = ctrl, tuneLength = 1)

но R продолжает падать. Как я могу улучшить производительность этой функции?

Answer 1

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

Вы можете попробовать установить для параметра k низкое значение и посмотреть, работает ли он:

knnFit <- train(
  Class ~ ., 
  data = ScaniaTrain, 
  method = "knn", 
  trControl = ctrl,
  tuneGrid = c(k=3)
)