提高机器学习预测速度计算

如何解决提高机器学习预测速度计算

我有一个 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 放在开头。