h2o - 来自 DRFModel 的 H2O Mojo 模型

Question

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

Answer 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));