Question

我正在使用h2o绑定进行一些DRF训练，并且代码是从h2o中的示例复制而来的。

    GBMParametersV3 gbmParams = new GBMParametersV3();
    gbmParams.trainingFrame = H2oApi.stringToFrameKey("train");
    gbmParams.validationFrame = H2oApi.stringToFrameKey("test");

    ColSpecifierV3 responseColumn = new ColSpecifierV3();
    responseColumn.columnName = ATT_LABLE_IRIS;
    gbmParams.responseColumn = responseColumn;

    GBMV3 gbmBody = h2o.train_gbm(gbmParams);

    JobV3 job = h2o.waitForJobCompletion(gbmBody.job.key);

    ModelKeyV3 modelKey = (ModelKeyV3)job.dest;
    ModelsV3 models = h2o.model(modelKey);
    LogService.getRoot().log(Level.INFO,"models: " + models);

    GBMModelV3 model = (GBMModelV3)models.models[0];

但是GBMModelV3在源代码中没有POJO和MOJO。如我们所见，它显示了this.havePojo = false;

public class GBMModelV3 extends SharedTreeModelV3<GBMParametersV3, GBMModelOutputV3> {
public GBMModelV3() {
    this.checksum = 0L;
    this.algo = "";
    this.algoFullName = "";
    this.responseColumnName = "";
    this.timestamp = 0L;
    this.havePojo = false;
    this.haveMojo = false;
}

public String toString() {
    return (new Gson()).toJson(this);
}

}

那么如何使用h2o-bindings api获得MOJO模型？或者如何在GBModelV3中获取树信息？因为我得到的GBMModelV3除了最终的树数据以外，还有很多参数。

我想要获得MOJO模型的原因是有人告诉我可以使用h2o-genmodel计算“ mojo_model.zip”并获取树。但是水绑定，该怎么做？

我可以使用h2o-bindings api获得MOJO模型吗？

0 个答案: