Question

我正在尝试使用函数:id保存模型构建，基于H2O user manual for R的第159页上的函数说明，参数仅考虑h2o.saveModel()。我查看了其他类似的函数，例如：path，但它使用相同的参数。请告知是否有其他方法来指定模型的名称。

Answer 1

模型文件的名称将由模型的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)

Answer 2

我认为更好的解决方法是每次保存模型时都生成一个唯一的文件夹。加载模型时，路径下始终只有一个模型文件。

saved_model = os.path.join('UNIQUE_MODEL_PATH', os.listdir('UNIQUE_MODEL_PATH')[0])
loaded_model = h2o.load_model(saved_model)

Answer 3

这是一种可行的方法：

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)

如何使用R中的h2o包保存模型时指定文件名

3 个答案: