r - 使用插入符号包构建模型时如何跟踪进度？

Question

我正在尝试使用插入符号包中的训练函数来构建模型：

 model <- train(training$class ~ .,data=training, method = "nb")

训练集包含约2万个观察值，每个观察值具有100多个变量。我想知道从该数据集构建模型是否需要数小时或数天。

如何估计从数据训练模型所需的时间？使用插入符号包中的功能时，如何跟踪培训过程的进度？

Answer 1

假设您正在使用


扩展的调整参数网格（调整参数的所有组合）
以及您选择的重采样技术（交叉验证，自举等）


你可以设置

trainctrl <- trainControl(verboseIter = TRUE)

并将其设置在训练函数的trControl参数中以跟踪训练进度

model <- train(training$class ~ .,data=training, method = 'nb', trControl = trainctrl)

这将在每个重新采样阶段将进度输出到控制台，并允许您评估训练/参数调整的进度。

要估算总运行时间，您可以运行一次模型以查看运行时间，然后根据重采样方案和参数组合的数量乘以相应的总估算时间。这可以通过再次设置trainControl并将tuneLength设置为1来完成：

trainctrl <- trainControl(method = 'none')
model <- train(training$class ~ ., data = training, method = 'nb', trControl = trainctrl, tuneLength = 1)

希望这可以帮助！ :)