在Spring Boot应用程序中获取h2o MOJO模型（zip文件）

Question

我正在编写一个Spring Boot应用程序（java类），它调用另一个java类中的H2o predict方法。

我已经独立测试了这个java类，如果MOJO模型位于调用java的同一位置，我就可以读取它，并且可以进行预测。

使用Spring boot App，这一次我的maven项目编译的结果jar无法读取该zip。我使用标准的方式使用h2o提供的包装器读取MOJO zip文件。

EasyPredictModelWrapper model = new EasyPredictModelWrapper(MojoModel.load("DRF_model_python_1504558159175_1.zip"));

File DRF_model_python_1504558159175_1.zip cannot be found.

我认为解决办法可以是：1。在Maven pom文件中添加一些内容，以便生成的jar知道从何处获取模型。或者2。如果MojoModel.Load方法接受文件的路径而不仅仅是文件名。但我觉得这行不通。

有什么想法吗？

Answer 1

您需要改变在项目中加载MOJO zip文件的方式，而需要将MOJO文件作为流从资源加载。您可以遵循这个示例，在这里您可以看到如何将MOJO放入资源文件夹，然后调用addMOJOsFromJARResource来包含MOJO。

项目:https://github.com/h2Oai/h2O-tutorials/tree/f67765bc6c68c2058d4b2786d1bbc627d3b70539/tutorials/hive_udf_template/hive_udf_mojo_template

1. Mojo zip存储在src/main/resources/model
2. 您需要如下所示引用h2o-genmodel类到

```

 import hex.genmodel.MojoReaderBackendFactory;
 import static hex.genmodel.MojoReaderBackendFactory.CachingStrategy;
 import hex.genmodel.MojoReaderBackend;
 import hex.genmodel.ModelMojoReader;

```

  public void addMOJOsFromJARResource() {
      try {
        String[] mojo_names = this.getMOJONames();
        for (int i = 0; i < mojo_names.length; i++) {
            MojoReaderBackend reader =
                MojoReaderBackendFactory.createReaderBackend(
                  getClass().getResourceAsStream(
                     "/models/"+mojo_names[i]), 
                      MojoReaderBackendFactory.CachingStrategy.MEMORY);
            MojoModel model = ModelMojoReader.readFrom(reader);
            this.addModel(model);
        }
       } catch (Exception e) {
         e.printStackTrace();
        throw new RuntimeException();
        }
    }