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)
)

train function in R caret package keeps crashing