来自DRFModel的H2O Mojo模型

时间:2017-05-18 19:15:36

标签: h2o sparkling-water

在scala中有一个训练有素的DRFModel实例,生成相应的MojoModel对象进行评分的最佳方法是什么?从我到目前为止看到的api,主要是围绕导出到一个文件,然后使用MojoModel.load(路径)加载备份,但没有直接转换?

1 个答案:

答案 0 :(得分:1)

当前模型实例无法通过MojoWriter转换为mojo实例。

MojoWriter提供方法

abstract public void writeTo(OutputStream os);

您可以使用它将mojo写入字节数组(使用ByteArrayOutputStream),然后将其用作mojo数据的源:

  ByteArrayOutputStream os = new ByteArrayOutputStream();
  model.getMojo().writeTo(os);
  MojoModel mojoModel = MojoModel.load(MojoReaderBackendFactory.createReaderBackend(
          new ByteArrayInputStream(os.toByteArray()), MojoReaderBackendFactory.CachingStrategy.MEMORY));