h20.ai JAVA REST API CLASS ISSUE二项式预测

时间:2017-11-07 10:24:38

标签: java r machine-learning deep-learning h2o

我正在使用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类。

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

1 个答案:

答案 0 :(得分:1)

有关如何使用POJO的文档,请访问:

以下示例来自文档。您可以看到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);