H2O有这么多子模块,它们之间是什么关系?
我使用h2o-bindings创建了一个流,但无法从结果中获取MOJO模型。
我搜索了答案,发现我必须添加h2o-genmodel的依赖项。 但是我仍然无法从h2o-bindings创建的结果中获得mojo模型。
那么水绑定和水基因模型之间的关系是什么? 如果我想训练数据,获取mojo模型并使用它在Java中构建树,该怎么办?
这是我使用h2o-bindings的代码。
@Override
ModelMetricsListSchemaV3 train(H2oApi h2o) throws IOException {
DRFParametersV3 drfParams = new DRFParametersV3();
drfParams.trainingFrame = H2oApi.stringToFrameKey("train");
drfParams.validationFrame = H2oApi.stringToFrameKey("test");
drfParams.ntrees=3;
ColSpecifierV3 responseColumn = new ColSpecifierV3();
responseColumn.columnName = ATT_LABEL_GOLF;
drfParams.responseColumn = responseColumn;
DRFV3 drfBody = h2o.train_drf(drfParams);
JobV3 job = h2o.waitForJobCompletion(drfBody.job.key);
ModelKeyV3 modelKey = (ModelKeyV3)job.dest;
ModelsV3 models = h2o.model(modelKey);
DRFModelV3 model = (DRFModelV3)models.models[0];
//I want to get mojo model but the output isn't right.
ModelExportV3 modelExportV3 = h2o.exportMojo(modelKey);
ModelMetricsListSchemaV3 predictParams = new ModelMetricsListSchemaV3();
predictParams.model = modelKey;
predictParams.frame = drfParams.trainingFrame;
predictParams.predictionsFrame = H2oApi.stringToFrameKey("predictions");
System.out.println(predictParams);
return predictParams;
}