Какова связь h2o-bindings и h2o-genmodel - PullRequest
Новая
       9

Какова связь h2o-bindings и h2o-genmodel

0 голосов
/ 31 августа 2018

H2O имеет так много подмодулей, как они связаны?

Я создал поток, используя привязки h2o, но не могу получить модель MOJO из результата.

Я искал ответ и обнаружил, что должен добавить зависимость h2o-genmodel. Но я все еще не могу получить модель mojo из результата, созданного h2o-bindings.

Так, какова связь h2o-связываний и h2o-genmodel? Что мне делать, если я хочу тренировать свои данные, получить модель mojo и использовать ее для построения дерева в java?

Это мой код с использованием h2o-привязок. 

@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;
}
...