r语言 - h20.ai 具有二项式预测的 JAVA REST API 类问题

Question

我正在使用 h2o 二项式预测并像这样转换几个字符串列

X2 <- as.numeric(as.factor(test$X2))
X3 <- as.numeric(as.factor(test$X3))
X4 <- as.numeric(as.factor(test$X4))

我为Rest API调用的二项式模型生成了h2o java pojo类。

那么，如何在 Java 中将我的 2 个字符串列转换为所需的格式。因为，h2o团队提到所有转换都应该在h20 api调用之前完成。

Answer 1

有关如何使用 POJO 的文档可以在这里找到：

• http://docs.h2o.ai/h2o/latest-stable/h2o-genmodel/javadoc/index.html

以下示例取自文档。 可以看到 RowData 对象包含要预测的新数据点的名称和值。 您只需传入字符串值，EasyPredictModelWrapper 就会将它们转换为模型知道如何使用的内容。

String modelClassName = "your_pojo_model_class_name";
hex.genmodel.GenModel rawModel;
rawModel = (hex.genmodel.GenModel) Class.forName(modelClassName).newInstance();
EasyPredictModelWrapper model = new EasyPredictModelWrapper(rawModel);
RowData row = new RowData();
row.put("Year", "1987");
row.put("Month", "10");
row.put("DayofMonth", "14");
row.put("DayOfWeek", "3");
row.put("CRSDepTime", "730");
row.put("UniqueCarrier", "PS");
row.put("Origin", "SAN");
row.put("Dest", "SFO");
BinomialModelPrediction p = model.predictBinomial(row);