将逻辑回归LBFGS模型转换为POJO

Question

看看H20，我想知道是否有办法将Spark LogisticRegressionWithLBFGS（）模型转换为POJO对象。 目前，我一直使用LogisticRegressionModel.load（）函数加载它，但它需要大约40秒。

Answer 1

目前H2O不支持将Spark模型导出到POJO中。但是，您始终可以使用Sparkling Water：在Spark中准备数据，然后调用val trainHF = h2oContext.asH2OFrame(sparkDataFrame)并在该数据上构建GLM模型。 H2O的GLM模型可以很容易地导出到POJO。

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