保存随机的Forest对象

时间:2011-10-23 02:21:39

标签: r

我有一个randomForest对象,我想保存供以后使用。我尝试过以下一些但没有运气。

save(topDawg , file="myRFobject.RData")

这只保存字符串“topDawg”

> formula(topDawg)
Error in formula.default(topDawg) : invalid formula


> save(getTree(topDawg))
Error in save(getTree(topDawg)) : object ‘getTree(topDawg)’ not found

有什么建议吗?

4 个答案:

答案 0 :(得分:24)

我不确定你要在这里做什么,因为通常你save一个对象,然后再load,就像这样:

set.seed(71)
> irisrf <- randomForest(Species ~ ., data=iris, importance=TRUE,
+                         proximity=TRUE)
> save(irisrf,file = "irisrf.RData")
> 
> rm(irisrf)
> print(irisrf)
Error in print(irisrf) : object 'irisrf' not found
> 
> load("irisrf.RData")
> print(irisrf)

Call:
 randomForest(formula = Species ~ ., data = iris, importance = TRUE,      proximity = TRUE) 
               Type of random forest: classification
                     Number of trees: 500
No. of variables tried at each split: 2

        OOB estimate of  error rate: 4.67%
Confusion matrix:
           setosa versicolor virginica class.error
setosa         50          0         0        0.00
versicolor      0         47         3        0.06
virginica       0          4        46        0.08

答案 1 :(得分:4)

我有同样的问题(加载RandomForest对象导致字符串),这样的东西似乎对我有用:

inlink->next

(我有一个随机的森林对象&#39; forestGOOG&#39;保存在工作目录中)

答案 2 :(得分:1)

如果您想以其他名称

加载模型,这是一个解决方案
  library(randomForest)

  # 1. Create data set
  set.seed(100)
  df_iris <- randomForest(Species ~ ., data = iris, importance = TRUE,  proximity = TRUE)

  # 2. Save model
  file_name <- "model_iris.rds"
  saveRDS(df_iris, file_name)

  # 2.3. Load model under another name
  df_iris_loaded <- readRDS(file_name)
  df_iris_loaded

  # 2.4. Test two models
  identical(df_iris, df_iris_loaded, ignore.environment = TRUE)

答案 3 :(得分:0)

保存和读取随机森林模型

#model
  model_rf <- randomForest(y~ ., data = dataset, mtry=7,ntree=500)

#write model
  saveRDS(model_rf , "model_rf.RDS")

#load model
 readRDS("model_rf.RDS")