在 R 中保存和加载模型

Posted 2023-03-12

【中文标题】在 R 中保存和加载模型

【英文标题】：Saving and loading a model in R

【发布时间】：2013-01-23 13:33:57

【问题描述】：

使用caret 时，如何在训练后保存模型，并在稍后（例如在不同的会话中）加载它以进行预测？

【参考方案1】：

以下代码假定您的模型的变量名称是“模型”：

save(model, "model.RData")

这会将您的模型保存为当前工作目录中的“model.RData”。您可以通过发出以下命令找出工作目录是什么：

getwd()

要重新加载它，请确保您的模型保存在您的工作目录中并发出：

load("model.RData")

【讨论】：

我需要通过save(model, file = "model.RData")显式设置文件

【参考方案2】：

正确的语法是：

save(model, file="model.Rdata")

之后，可以使用 load() 命令加载它。

【讨论】：

嗯，是的，有趣的是，另一个答案两年半没有改变并且不正确。哈哈

【参考方案3】：

现在更好的解决方案是使用 saveRDS 保存和 readRDS 读取：

saveRDS(model, "model.rds")
my_model <- readRDS("model.rds")

这使您可以为对象选择一个新名称（您无需记住保存时使用的名称）