【问题标题】:Improve machine learning prediction speed computation提高机器学习预测速度计算
【发布时间】:2021-04-06 04:46:29
【问题描述】:

我有一个 knn 模型自 7 天以来仍在计算。这正常吗??? 我的火车数据集有 983669 行和两个连续预测变量和一个分类变量(DV 是连续的)。有没有办法提高速度?

我的联想“军团”拥有英特尔酷睿 i7(第 9 代,2.6 GHz),16GB RAM,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")

【问题讨论】:

    标签: r machine-learning data-mining r-caret


    【解决方案1】:

    您可以尝试使用 Google Colab (colab.research.google.com)。在那里,您可以将运行时更改为 GPU 或 TPU 并提高模型速度。 另外,将 repeats=1 放在开头。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2016-11-16
      • 1970-01-01
      • 2017-05-09
      • 1970-01-01
      • 1970-01-01
      • 2019-09-23
      • 2021-05-26
      • 2021-04-09
      相关资源
      最近更新 更多