【发布时间】:2018-07-27 17:40:03
【问题描述】:
我正在尝试使用函数:h2o.saveModel() 保存模型构建,基于 H2O user manual for R 第 159 页上的函数描述,参数仅考虑 path。我查看了其他类似的函数,例如:h2o.saveModelDetails(),但它使用相同的参数。请告知是否有其他方法可以指定模型的名称。
【问题讨论】:
-
可以重命名模型的model_path
标签: h2o
我正在尝试使用函数:h2o.saveModel() 保存模型构建,基于 H2O user manual for R 第 159 页上的函数描述,参数仅考虑 path。我查看了其他类似的函数,例如:h2o.saveModelDetails(),但它使用相同的参数。请告知是否有其他方法可以指定模型的名称。
【问题讨论】:
标签: h2o
在python中:
model_path = h2o.save_model(model=model, path="mymodel1", force=True)
path = os.path.dirname(os.path.abspath(model_path))
os.rename(model_path, os.path.join(path,f'h2o_new_name'))
【讨论】:
我认为更好的解决方法是每次保存模型时生成一个唯一文件夹。加载模型时,路径下总是只有一个模型文件。
saved_model = os.path.join('UNIQUE_MODEL_PATH', os.listdir('UNIQUE_MODEL_PATH')[0])
loaded_model = h2o.load_model(saved_model)
【讨论】:
这是一种可能的方法:
output_dir <-getwd()
DRF_MO <- h2o.saveModel(object=aml, path=output_dir, force=TRUE)
DRF_MO <- file.path(output_dir, aml@algorithm)
file.rename(file.path(output_dir, aml@model_id), DRF_MO)
【讨论】:
DRF_MO)
模型文件的名称将由模型的 ID 决定。因此,如果您在训练模型时指定model_id,则可以对其进行自定义。目前无法在模型训练后更改其 ID。
文件保存后可以重命名:
h2o.saveModel(object = fit, path = path.value, force = TRUE) # force overwriting
name <- file.path(path.value, fileName) # destination file name at the same folder location
file.rename(file.path(path.value, fit@model_id), name)
【讨论】: