如何保存parsnip模型拟合(来自ranger)？

Question

我有一个parsnip模型(来自ranger)，大致来自here

# install.packages("tidymodels")

data(cells, package = "modeldata")

rf_mod <- 
  rand_forest(trees = 100) %>% 
  set_engine("ranger") %>% 
  set_mode("classification")

set.seed(123)
cell_split <- initial_split(cells %>% select(-case), strata = class)

cell_train <- training(cell_split)

rf_fit <- 
  rf_mod %>% 
  fit(class ~ ., data = cell_train)

> class(rf_fit)
[1] "_ranger"   "model_fit"

如何将其保存到磁盘，以便稍后加载？

我尝试过dput，但得到了一个错误：

dput(rf_fit, file="rf_fit.R")
rf_fit2 <- dget("rf_fit.R")
Error in missing_arg() : could not find function "missing_arg"

这是真的，model_fit.R文件中有几个missing_arg调用，这似乎是一种标记缺少参数的方式。然而，这只是一条副线。我不需要使用dput，我只希望能够保存和加载模型。

Answer 1

尝试使用此选项。save()和load()函数允许您存储模型，然后再次添加墨迹。代码如下：

data(cells, package = "modeldata")

rf_mod <- 
  rand_forest(trees = 100) %>% 
  set_engine("ranger") %>% 
  set_mode("classification")

set.seed(123)
cell_split <- initial_split(cells %>% select(-case), strata = class)

cell_train <- training(cell_split)

rf_fit <- 
  rf_mod %>% 
  fit(class ~ ., data = cell_train)

#Export option
save(rf_fit,file='Mymod.RData')
load('Mymod.RData')

另一种选择是使用saveRDS()保存模型，然后使用readRDS()加载模型，但需要在对象中进行分配：

#Export option 2
saveRDS(rf_fit, file = "Mymod.rds")
# Restore the object
rf_fit <- readRDS(file = "Mymod.rds")

Answer 2

正如Duck提到的，saveRDS()和readRDS()可以用来保存/加载任何R对象。同样，save()和load()也可以用于相同的目的。有许多比较这两种方法的在线讨论/博客。