在R中保存和加载模型

42

当使用caret时,我如何在训练后保存模型,并在以后的时间(例如在不同的会话中)加载它进行预测?

3个回答

62

现在更好的解决方案是使用saveRDS进行保存,使用readRDS进行读取:

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

这允许您选择一个新名称来命名对象(您无需记住保存时使用的名称)。


23

正确的语法是使用:

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

之后,可以使用load()命令进行加载。


1
嗯,是的,有趣的是另一个答案在两年半内没有改变且不正确。哈哈 - Rich Scriven

11
以下代码假定你的模型变量名为“model”:
save(model, "model.RData")

这将在当前工作目录中保存您的模型为 "model.RData"。您可以通过执行以下命令来查找工作目录:

getwd()

要重新加载模型,请确保您的模型保存在工作目录中,然后执行以下操作:

load("model.RData")

3
在我的情况下,我需要通过 save(model, file = "model.RData") 显式地设置文件。 - Boern

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接