问题描述
我有一个 knn 模型在 7 天后仍在计算。这正常吗??? 我的训练数据集有 983669 行和两个连续预测变量和一个分类变量(DV 是连续的)。有没有办法提高速度?
我的联想“军团”拥有英特尔酷睿 i7(第 9 代,2.6 GHz)、16GB 内存、2070 Nvidia RTX、64 位
参数设置为
cv <- trainControl(method = "repeatedcv",number = 10,repeats = 5)
hypergrid <- expand.grid(k = seq(2,30,by = 0.8)
knn <- train(DV ~.,data = train,method = "knn",trControl = cv,tuneGrid = hypergrid,metric = "RMSE")
解决方法
您可以尝试使用 Google Colab (colab.research.google.com)。在那里,您可以将运行时更改为 GPU 或 TPU 并提高模型速度。 另外,将repeats=1 放在开头。