2016-02-05 77 views
0

看着H20,我想知道是否有方法可以将Spark LogisticRegressionWithLBFGS()模型转换为POJO对象。 目前,我一直使用LogisticRegressionModel.load()函数加载它,但大约需要40秒。将逻辑回归LBFGS模型转换为POJO

回答

0

现在H2O不支持将Spark模型导出到POJO中。但是,您始终可以使用Sparkling Water:在Spark中准备数据,然后致电val trainHF = h2oContext.asH2OFrame(sparkDataFrame)并在该数据上构建GLM模型。 H2O的GLM模型可以轻松导出到POJO。

或者通过一些努力,您可以从Spark的GLM模型中提取信息并将其填充到H2O的GLM模型中。