来自 DRFModel 的 H2O Mojo 模型

Question

在 scala 中有一个经过训练的 DRFModel 实例，生成相应的 MojoModel 对象以进行评分的最佳方法是什么？从我目前看到的 api s 来看，主要是导出到一个文件，然后使用 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));

来自 DRFModel 的 H2O Mojo 模型

H2O Mojo model from DRFModel

h2o

sparkling-water