【问题标题】:How to save All models from h2o automl如何从 h2o automl 中保存所有模型
【发布时间】:2019-02-17 07:29:44
【问题描述】:

我正在尝试将h2o.automl 中的所有模型保存为h2o 包的一部分。目前我可以使用h2o.saveModel(aml@leader, path = "/home/data/user") 保存单个模型。

如何保存所有模型?

这是我对示例数据集的尝试:

 library(h2o)
 h2o.init()
 prostate.hex <- h2o.importFile(path = paste("https://raw.github.com",
    "h2oai/h2o-2/master/smalldata/logreg/prostate.csv", sep = "/"),
    destination_frame = "prostate.hex")

从github获取数据或通过readr导入:

 library(readr)
 prostate <- read_csv("/home/data/user/prostate.csv")

 prostate.hex<- as.h2o(prostate, "prostate.hex")

 aml <- h2o.automl(y = "CAPSULE", x = c("AGE","RACE","PSA","DCAPS"),
    training_frame = prostate.hex,
    max_runtime_secs = 180,
    exclude_algos = c("StackedEnsemble")
    )

现在我正在尝试将模型保存在aml

mod_ids <- as_tibble(aml@leaderboard$model_id)

现在我不知道如何保存模型:

 for(i in 1:nrow(mod_ids)) {
   print(mod_ids[i,])
   #h2o.saveModel(object = aml@leaderboard[[i]], "/home/data/user/")
 }

这是我尝试过的:

tutorial automl

H2O AUTOML: How to save reuse and build on top of existing automl models

【问题讨论】:

    标签: r h2o


    【解决方案1】:

    试试这个,它会做你的工作:

     for(i in 1:nrow(mod_ids)) {
    
        aml1 <- h2o.getModel(aml@leaderboard[i, 1]) # get model object in environment
        h2o.saveModel(object = aml1, "C:/Users/sm/Documents/stack/models") # pass that model object to h2o.saveModel as an argument
    
      }
    

    【讨论】:

      猜你喜欢
      • 2021-08-08
      • 2018-06-16
      • 2019-10-11
      • 2020-02-05
      • 2021-09-18
      • 2018-01-21
      • 1970-01-01
      • 2019-10-04
      • 1970-01-01
      相关资源
      最近更新 更多